约会安排 - HDU 4553 - 线段树 - 区间合并

约会安排 - HDU 4553 - 线段树 - 区间合并

思路

  国际惯例中文题目直接说思路，DS QT：找一段最靠前长度为QT的空间；NS QT：找一段最靠前长度为QT的空间，如果没有找到可以将DS占据的空间当做空闲空间，找一段最靠前的空间；STUDY!! L R：清空[L,R]区间的空间。

  用两个线段树，一个记录DS，一个记录NS，维护最大左区间ma_l，最大右区间ma_r，以及区间内最大区间ma。

  判断是否可以容下一端长为qt的空间可以直接通过比较根节点的ma和qt的大小就可以。因为ma记录的是当前区间的最大空闲区间，也就是说只需要比较qt和tree[1].ma大小即可判断空间是否足够。

代码

//
//  main.cpp
//  L
//
//  Created by LucienShui on 2017/6/4.
//  Copyright © 2017年 LucienShui. All rights reserved.
//

#include <iostream>
#include <algorithm>
#include <set>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <iomanip>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cctype>
#define memset(a,b,n) memset(a,b,n*sizeof(a[0]))
#define il inline
#define ll long long
#define ull unsigned long long

using namespace std;

#define ls (u<<1)
#define rs (u<<1|1)
#define lson u<<1,l,mid
#define rson u<<1|1,mid+1,r
#define maxn 100007

struct node
{
    int l , r ;
    int ma_l ;
    int ma_r ;
    int ma ;
    int lazy ;
}tree[2][maxn<<2] ;
void build(int l , int r , int u)
{
    for(int i = 0;i < 2;i++)
    {
        tree[i][u].l = l ;
        tree[i][u].r = r;
        tree[i][u].lazy = -1;
        tree[i][u].ma_l = tree[i][u].ma = tree[i][u].ma_r  = (r - l + 1) ;
    }
    if(l == r)return ;
    int mid = (l + r)>>1 ;
    build(l , mid , ls) ;
    build(mid+1 , r , rs) ;
}
void push_down(int u)
{
    for(int i = 0 ;i < 2;i++)
        if(tree[i][u].lazy != -1)
        {
            tree[i][ls].ma_l = tree[i][ls].ma_r = tree[i][ls].ma = (tree[i][ls].r - tree[i][ls].l + 1)*tree[i][u].lazy ;
            tree[i][rs].ma_l = tree[i][rs].ma_r = tree[i][rs].ma = (tree[i][rs].r - tree[i][rs].l + 1)*tree[i][u].lazy ;
            tree[i][ls].lazy = tree[i][rs].lazy  = tree[i][u].lazy ;
            tree[i][u].lazy = -1;
        }
}
void push_up(int u)
{
    for(int i = 0 ;i < 2;i++)
    {
        tree[i][u].ma_l = tree[i][ls].ma_l ;
        tree[i][u].ma_r = tree[i][rs].ma_r ;
        tree[i][u].ma = max(max(tree[i][ls].ma ,tree[i][rs].ma) ,tree[i][ls].ma_r+tree[i][rs].ma_l) ;
        if(tree[i][u].ma_l == (tree[i][ls].r - tree[i][ls].l + 1))
            tree[i][u].ma_l += tree[i][rs].ma_l ;
        if(tree[i][u].ma_r == (tree[i][rs].r - tree[i][rs].l + 1))
            tree[i][u].ma_r += tree[i][ls].ma_r ;
    }
}
int query(int u , const int flag , const int qt)
{
    if(tree[flag][u].l == tree[flag][u].r)//如果到达叶节点
        return tree[flag][u].l ;//返回叶节点的坐标
    push_down(u) ;
    if(tree[flag][ls].ma >= qt)//优先向前安排，所以优先询问左子树是否可以容纳下qt
        return  query(ls , flag ,qt) ;
    else if(tree[flag][ls].ma_r + tree[flag][rs].ma_l >= qt)
        //询问中间，左子树的右区间加上右子树的左区间是否可以容纳下qt
        return   tree[flag][ls].r - tree[flag][ls].ma_r + 1 ;
        //如果可以容纳下，那么左子树的右区间的左端点就一定是可以容纳下qt的第一个区间的左端点
    else return query(rs , flag , qt) ;//最坏的情况下判断右子树是否可以容纳下qt
}
void update(int a , int b , int u , const int flag , const int tag)
{
    if(tree[flag][u].l >= a && tree[flag][u].r <= b)
    {
        tree[flag][u].lazy = tag;
        tree[flag][u].ma_l = tree[flag][u].ma = tree[flag][u].ma_r = (tree[flag][u].r  - tree[flag][u].l + 1)*tag ;
        return ;
    }
    push_down(u) ;
    int mid = (tree[flag][u].l + tree[flag][u].r) >> 1 ;
    if(a <= mid)update(a , b , ls , flag , tag) ;
    if(b > mid)update(a , b , rs , flag , tag) ;
    push_up(u) ;
}

int main ()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
#endif // ONLINE_JUDGE
    int T ;
    scanf("%d" , &T) ;
    int cas = 0 ;
    while(T--)
    {
        int N , Q ;
        scanf("%d%d" ,&N , &Q) ;
        build(1 , N , 1) ;
        char ch[10] ;
        int L , R , qt ;
        printf("Case %d:\n" , ++cas) ;
        while(Q--)
        {
            scanf("%s"  , ch) ;
            if(ch[0] == 'D')
            {
                scanf("%d" , &qt) ;
                if(tree[1][1].ma < qt)
                {
                    puts("fly with yourself");
                    continue ;
                }
                int pos = query(1 , 1 , qt) ;
                update(pos , pos + qt - 1 , 1 , 1 , 0) ;
                printf("%d,let's fly\n" , pos) ;
            }
            else if(ch[0] == 'N')
            {
                scanf("%d" ,  &qt) ;
                int pos ;
                if(tree[1][1].ma < qt && tree[0][1].ma < qt) {
                    puts("wait for me");
                    continue ;
                }
                else if(tree[1][1].ma >= qt)
                    pos = query(1 , 1 , qt) ;
                else pos = query(1 , 0 , qt) ;
                printf("%d,don't put my gezi\n"  , pos) ;
                update(pos , pos + qt - 1, 1, 0, 0) ;
                update(pos , pos + qt - 1, 1, 1, 0) ;
            }
            else if(ch[0] == 'S')
            {
                scanf("%d%d" , &L , &R) ;
                update(L , R , 1 , 0 , 1) ;
                update(L , R , 1 , 1 , 1) ;
                puts("I am the hope of chinese chengxuyuan!!") ;
            }
        }
    }
    return  0 ;
}