hdu 4266 The Worm in the Apple (线段树)

/*
题意：给你N个花瓶，编号是0  到 N - 1 ，初始状态花瓶是空的，每个花瓶最多插一朵花。
然后有2个操作。
操作1，a b c ，往在a位置后面(包括a)插b朵花，输出插入的首位置和末位置。
操作2,a b ,输出区间[a , b ]范围内的花的数量，然后全部清空。


很显然这是一道线段树。区间更新，区间求和，这些基本的操作线段树都可以logN的时间范围内完成。
操作2，很显然就是线段树的区间求和，求出[a , b]范围内的花朵的数量，区间更新，将整个区间全部变成0。
操作1，这里我们首先需要找出他的首位置和末位置，所以需要二分他的位置。
首先我们二分他的首位置, l = a , r = n ，在这个区间内二分，找出第一个0的位置，那就是该操作的首位置pos1。
然后再二分他的末位置，l = pos1 , r = n ，找到第b个0，就是该操作的末位置pos2，然后区间更新[pos1 ,pos2]全部置为1。
*/
#include <cstdio>
# include <algorithm>
using namespace std;
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
///lson和rson分别表示结点的左儿子和右儿子
///rt表示当前子树的根(root),也就是当前所在的结点
const int maxn =60010;
///maxn是题目给的最大区间,而节点数要开4倍,确切的来说节点数要开大于maxn的最小2x的两倍
int sum[maxn<<2];///求和
int col[maxn<<2];///用来标记每个节点，为0则表示没有标记，否则为标记
int str[maxn];
void PushUP(int rt)///PushUP(int rt)是把当前结点的信息更新到父结点
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void PushDown(int  rt,int  m)///pushDown(int rt)是把当前结点的信息更新给儿子结点，m为分区间的长度
{
    if (col[rt])///被标记过，说明区间改变了
    {
        col[rt<<1] = col[rt<<1|1] = 1;
        sum[rt<<1] = (m - (m >> 1)) * col[rt];///更新左儿子的和
        sum[rt<<1|1] = (m >> 1) * col[rt];///更新右儿子的和
        col[rt] = -1;///将标记向儿子节点移动后，父节点的延迟标记去掉
        ///传递后，当前节点标记域清空
    }
    else
    {
        col[rt<<1] = col[rt<<1|1] = 0;
        sum[rt<<1] = sum[rt<<1|1] = 0;
        col[rt] = -1;
    }
}
void build(int l,int  r,int rt)///建树
{
    col[rt] = -1;
    if (l == r)
    {
        sum[rt]=0;
        //  scanf("%lld",&sum[rt]);
        return ;
    }
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    PushUP(rt);
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L <= l && r <= R)
    {
        col[rt]= c;
        sum[rt]= c *(r - l +1);
        return;
    }
    if(col[rt]!=-1)
        PushDown(rt , r - l +1);
    int m =(l + r)>>1;
    if(L <= m) update(L , R , c , lson);
    if(R > m) update(L , R , c , rson);
    PushUP(rt);
}
int query(int L,int R,int l,int r,int rt)///区间求和
{
    if (L <= l && r <= R)
    {
        return sum[rt];
    }
    if(col[rt]!=-1)
        PushDown(rt , r - l + 1);///要取rt子节点的值时，也要先把rt的延迟标记向下移动
    int m = (l + r) >> 1;
    int  ret = 0;
    if (L <= m) ret += query(L , R , lson);
    if (m < R) ret += query(L , R , rson);
    return ret;
}
int main()
{
    int t,m,n,a,b,c;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        build(1,n,1);
        while(m--)
        {
            scanf("%d",&a);
            if(a==1)
            {
                scanf("%d%d",&b,&c);
                b++;
                c=min(c,(n-b+1)-query(b,n,1,n,1));
                if(c==0)
                {
                    printf("Can not put any one.\n");
                    continue;
                }
                int left=b;
                int right=n;
                while(left<=right)
                {
                    int mid=(left+right)/2;
                    if(query(b,mid,1,n,1)<mid-b+1)
                        right=mid-1;
                    else
                        left=mid+1;
                }
                b=left;
                right=n;
                while(left<=right)
                {
                    int mid=(left+right)/2;
                    int ans1=query(b,mid,1,n,1);
                    int ans2=mid-b+1;
                    int ans=ans2-ans1;
                    if(ans<c)
                        left=mid+1;
                    else
                        right=mid-1;
                }
                if(left>n)
                    left=n;
                printf("%d %d\n",b-1,left-1);
                update(b,left,1,1,n,1);
            }
            else
            {
                scanf("%d%d",&b,&c);
                b++;
                c++;
                if(b>=n)
                    b=n;
                if(c>=n)
                    c=n;
                int ans=query(b,c,1,n,1);
                printf("%d\n",ans);
                update(b,c,0,1,n,1);
                // printf("%d\n",query(b,c,1,n,1));

            }
        }
        printf("\n");
    }
    return 0;
}
/*
10
10 4
1 2 9
1 0 9
2 0 9
2 0 1
10 5
1 3 5
2 4 5
1 1 8
2 3 6
1 8 8
10 6
1 2 5
2 3 4
1 0 8
2 2 5
1 4 4
1 2 3
*/