HDU-4553 约会安排（线段树维护连续区间）

http://acm.hdu.edu.cn/showproblem.php?pid=4553

 

Problem Description

寒假来了，又到了小明和女神们约会的季节。 
　　小明虽为屌丝级码农，但非常活跃，女神们常常在小明网上的大段发言后热情回复“呵呵”，所以，小明的最爱就是和女神们约会。与此同时，也有很多基友找他开黑，由于数量实在过于巨大，怎么安排时间便成了小明的一大心事。 
　　我们已知小明一共有T的空闲时间，期间会有很多女神或者基友来找小明。 
　　作为一个操作系统曾经怒考71分的大神，小明想到了一个算法，即“首次适应算法”，根据操作系统课本的描述，就是找一段最靠前的符合要求的连续空间分配给每个请求，由此小明做出了一个决定： 
　　当一个基友来找小明时，小明就根据“首次适应算法”来找一段空闲的时间来和基友约好，如果找到，就说“X,let’s fly”（此处，X为开始时间），否则就说“fly with yourself”；
　　当女神来找小明时，先使用一次“首次适应算法”，如果没有找到，小明就冒着木叽叽的风险无视所有屌丝基友的约定，再次使用“无视基友首次适应算法”，两次只要有一次找到，就说“X,don’t put my gezi”（此处，X为开始时间），否则就说“wait for me” 
　　当然，我们知道小明不是一个节操负无穷的人，如果和女神约会完，还有剩余时间，他还是会和原来约好的基友去dota的。（举个例子：小西（屌丝）和小明约好在1~5这个时间单位段内打dota，这时候，女神来和小明预约长度为3的时间段，那么最终就是1~3小明去和女神约会，搞定后在4~5和小西打dota） 
　　小明偶尔也会想要学习新知识，此时小明就会把某一个时间区间的所有已经预定的时间全部清空用来学习并且怒吼“I am the hope of chinese chengxuyuan!!”，不过小明一般都是三分钟热度，再有人来预定的话，小明就会按耐不住寂寞把学习新知识的时间分配出去。

Input

输入第一行为CASE，表示有CASE组测试数据； 
每组数据以两个整数T，N开始，T代表总共的时间，N表示预约请求的个数； 
接着的N行，每行表示一个女神或者基友的预约，“NS QT”代表一个女神来找小明约一段长为QT的时间，“DS QT”则代表一个屌丝的长为QT的请求，当然也有可能是小明想学知识了，“STUDY!! L R”代表清空L~R区间内的所有请求。 

[Technical Specification] 
1. 1 <= CASE <= 30 
2. 1 <= T, N <= 100000 
3. 1 <= QT <= 110000 
4. 1 <= L <= R <=T

Output

对于每一个case，第一行先输出“Case C:”代表是第几个case，然后N行，每行对应一个请求的结果(参照描述)。 
输出样本(可复制此处)： 
“X,let's fly”,”fly with yourself”,”X,don't put my gezi”,”wait for me”,”I am the hope of chinese chengxuyuan!!” 

Sample Input

1
5 6
DS 3
NS 2
NS 4
STUDY!! 1 5
DS 4
NS 2

Sample Output

Case 1:
1,let's fly
4,don't put my gezi
wait for me
I am the hope of chinese chengxuyuan!!
1,let's fly
1,don't put my gezi

 

题意：

也就是一个优先级问题，学习优先级最高，其次约女神，约屌丝最低

思路：

使用2棵线段树，维护最长空闲时间段，一颗(A)既有屌丝又有女神，表示约人，另一颗(B)只有女神，表示约女神（忽视基友的存在）。

若有基友约，直接判断A是否还有位置，有就更新A区间。

若有女神约，则先判断A是否有位置，有就同时更新区间A和B，否则再判断B是否有位置(忽视约屌丝)，若有还是同时更新区间A和B。

这里的标记我是直接用 最长连续区间的值表示，因为答应基友或者女神，那么有一段的最长连续区间肯定为0。

如果要学习 那么这一段的最长连续区间恢复成这一段的长度。

 

反思：

刚开始一直wa，不知道为啥，改BUG改了几个小时，绝望时看了oj平台上的comment，

看到了这两个评论：

Re:题目里把英文引号写成了中文，注意！！！！

Re:注意题目中的单引号，他给的是中文，记得改成英文！！！！不然会WA到哭。。。。。。。。。

wa到哭了，都怪自己没注意，复制的题目中的话，而不是样例输出中的话，哎，长记性了，以后代码的输出要在给的标准输出里面复制，不要再在文章里面复制了。

 

代码如下：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <string>
#include <math.h>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <stack>
#include <map>
#include <math.h>
const int INF=0x3f3f3f3f;
typedef long long LL;
const int mod=1e9+7;
//const double PI=acos(-1);
const int maxn=1e5+10;
using namespace std;
//ios::sync_with_stdio(false);
//    cin.tie(NULL);

struct SegTree_node
{
    int l;
    int r;
    int ls;//ls 包含左端点的最长连续区间
    int rs;//rs 包含右端点的最长连续区间
    int ms;//ms 最长连续区间
}SegTree[2][maxn<<2];//SegTree[0]代表有人约的时间 ，SegTree[1]代表女神约的时间 

void Build(int l,int r,int rt)
{
    SegTree[0][rt].l=SegTree[1][rt].l=l;
    SegTree[0][rt].r=SegTree[1][rt].r=r;
    SegTree[0][rt].ls=SegTree[0][rt].rs=SegTree[0][rt].ms=r-l+1;
    SegTree[1][rt].ls=SegTree[1][rt].rs=SegTree[1][rt].ms=r-l+1;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    Build(l,mid,rt<<1);
    Build(mid+1,r,rt<<1|1);
}

void PushUp(int rt)
{
    int l=SegTree[0][rt].l;
    int r=SegTree[0][rt].r;
    SegTree[0][rt].ls=SegTree[0][rt<<1].ls;//左子树的左连续区间肯定是父亲节点左连续区间的一部分
    SegTree[0][rt].rs=SegTree[0][rt<<1|1].rs;//同理
    //父亲节点的最长连续区间要么是 左子树的最长连续区间 要么  右子树的最长连续区间  要么 左子树右连续+右子树左连续
    SegTree[0][rt].ms=max(SegTree[0][rt<<1].rs+SegTree[0][rt<<1|1].ls,max(SegTree[0][rt<<1].ms,SegTree[0][rt<<1|1].ms));
    //如果左子树的左连续长度为整个左子树的长度，那么父亲节点的左连续区间得加上 右子树的左连续区间
    if(SegTree[0][rt<<1].ls==SegTree[0][rt<<1].r-SegTree[0][rt<<1].l+1)
        SegTree[0][rt].ls+=SegTree[0][rt<<1|1].ls;
    if(SegTree[0][rt<<1|1].rs==SegTree[0][rt<<1|1].r-SegTree[0][rt<<1|1].l+1)//同理
        SegTree[0][rt].rs+=SegTree[0][rt<<1].rs;
    
    SegTree[1][rt].ls=SegTree[1][rt<<1].ls;
    SegTree[1][rt].rs=SegTree[1][rt<<1|1].rs;
    SegTree[1][rt].ms=max(SegTree[1][rt<<1].rs+SegTree[1][rt<<1|1].ls,max(SegTree[1][rt<<1].ms,SegTree[1][rt<<1|1].ms));
    if(SegTree[1][rt<<1].ls==SegTree[1][rt<<1].r-SegTree[1][rt<<1].l+1)
        SegTree[1][rt].ls+=SegTree[1][rt<<1|1].ls;
    if(SegTree[1][rt<<1|1].rs==SegTree[1][rt<<1|1].r-SegTree[1][rt<<1|1].l+1)
        SegTree[1][rt].rs+=SegTree[1][rt<<1].rs;
}

void PushDown(int rt)//下推标记
{
    int l=SegTree[0][rt].l;
    int r=SegTree[0][rt].r;
    int mid=(l+r)>>1;
    if(SegTree[0][rt].ms==r-l+1)//可能是学习操作造成的要下推
    {
        SegTree[0][rt<<1].ls=SegTree[0][rt<<1].rs=SegTree[0][rt<<1].ms=mid-l+1;
        SegTree[0][rt<<1|1].ls=SegTree[0][rt<<1|1].rs=SegTree[0][rt<<1|1].ms=r-mid;
    }
    else if(SegTree[0][rt].ms==0)//由约人操作造成
    {
        SegTree[0][rt<<1].ls=SegTree[0][rt<<1].rs=SegTree[0][rt<<1].ms=0;
        SegTree[0][rt<<1|1].ls=SegTree[0][rt<<1|1].rs=SegTree[0][rt<<1|1].ms=0;
    }
    
    if(SegTree[1][rt].ms==r-l+1)
    {
        SegTree[1][rt<<1].ls=SegTree[1][rt<<1].rs=SegTree[1][rt<<1].ms=mid-l+1;
        SegTree[1][rt<<1|1].ls=SegTree[1][rt<<1|1].rs=SegTree[1][rt<<1|1].ms=r-mid;
    }
    else if(SegTree[1][rt].ms==0)
    {
        SegTree[1][rt<<1].ls=SegTree[1][rt<<1].rs=SegTree[1][rt<<1].ms=0;
        SegTree[1][rt<<1|1].ls=SegTree[1][rt<<1|1].rs=SegTree[1][rt<<1|1].ms=0;
    }
}

void Update(int L,int R,int flag,int rt)
{
    int l=SegTree[0][rt].l;
    int r=SegTree[0][rt].r;
    if(L<=l&&R>=r)
    {
        if(flag==0)//约基友
        {
            SegTree[0][rt].ls=SegTree[0][rt].rs=SegTree[0][rt].ms=0;
        }
        else if(flag==1)//约女神，要同时修改两个区间 
        {
            SegTree[0][rt].ls=SegTree[0][rt].rs=SegTree[0][rt].ms=0;
            SegTree[1][rt].ls=SegTree[1][rt].rs=SegTree[1][rt].ms=0;
        }
        else//学习，要同时修改两个区间 
        {
            SegTree[0][rt].ls=SegTree[0][rt].rs=SegTree[0][rt].ms=r-l+1;
            SegTree[1][rt].ls=SegTree[1][rt].rs=SegTree[1][rt].ms=r-l+1;
        }
        return ;
    }
    PushDown(rt);
    int mid=(l+r)>>1;
    if(L<=mid)
        Update(L,R,flag,rt<<1);
    if(R>mid)
        Update(L,R,flag,rt<<1|1);
    PushUp(rt);
}

int Query(int flag,int QT,int rt)
{
    int l=SegTree[flag][rt].l;
    int r=SegTree[flag][rt].r;
    if(l==r)
        return l;
    PushDown(rt);
    int mid=(l+r)>>1;
    if(SegTree[flag][rt<<1].ms>=QT)
        return Query(flag,QT,rt<<1);
    else if(SegTree[flag][rt<<1].rs+SegTree[flag][rt<<1|1].ls>=QT)
        return mid-SegTree[flag][rt<<1].rs+1;
    else
        return Query(flag,QT,rt<<1|1);
}

int main()
{
    //freopen("sample.txt","r",stdin);
    int CASE;
    scanf("%d",&CASE);
    for(int k=1;k<=CASE;k++)
    {
        int T,N;
        scanf("%d %d",&T,&N);
        printf("Case %d:\n",k);
        Build(1,T,1);
        char str[10];
        while(N--)
        {
            scanf("%s",str);
            if(str[0]=='D')
            {
                int QT;
                scanf("%d",&QT);
                if(SegTree[0][1].ms>=QT)//看整个区间内的最长连续区间是否满足QT
                {
                    int t=Query(0,QT,1);
                    printf("%d,let's fly\n",t);
                    Update(t,t+QT-1,0,1);
                }
                else
                    printf("fly with yourself\n");
            }
            else if(str[0]=='N')
            {
                int QT;
                scanf("%d",&QT);
                if(SegTree[0][1].ms>=QT)
                {
                    int t=Query(0,QT,1);
                    printf("%d,don't put my gezi\n",t);
                    Update(t,t+QT-1,1,1);
                }
                else if(SegTree[1][1].ms>=QT)
                {
                    int t=Query(1,QT,1);
                    printf("%d,don't put my gezi\n",t);
                    Update(t,t+QT-1,1,1);
                }
                else
                    printf("wait for me\n");
            }
            else if(str[0]=='S')
            {
                int L,R;
                scanf("%d %d",&L,&R);
                Update(L,R,2,1);
                printf("I am the hope of chinese chengxuyuan!!\n");
            }
        }
    }
    return 0;
}

 

 

贴出参考的博客：
https://blog.csdn.net/dsaghjkye/article/details/81069500

https://www.xuebuyuan.com/3225923.html

 

另一种思路，单个线段树加屌丝、女神标记的解法：

https://blog.csdn.net/qq_36782366/article/details/75206287