HDU-4614 线段树

  Alice有N个花瓶（标号为0~ N-1）。当她收到一些花时，她会随机的选择一个瓶子A，从它开始遍历A，A+1, A+2, ..., N-1号瓶子，遇到空瓶子就放一朵花进去，直到花朵放完或没有瓶子，剩下的花将被丢弃。有时，她也会清理标号从A到B的花瓶(A <= B).花瓶里的花会被丢弃。

Input

　　第一行一个整数T，表示数据组数。
　　每组数据，第一行一个整数N(1 < N < 50001) and M(1 < M < 50001). N 是花瓶个数,  M 是Alice的操作次数. 接下来M行 行3个 整数. 第一个整数 K(1 or 2). 如果K=1, 后面跟两个整数 A 和  F . 表示Alice 得到了F 朵花并且把它们放入从A 的花瓶里. 如果K= 2, 后跟两个整数 A 和  B. 表示Alice  清理的花瓶标号范围(A <= B).

Output

　　对于每个K=1的操作，输出第一朵和最后一朵花放置的花瓶标号。如果没有任何放花的位置，输出'Can not put any one.'.对于K=2的操作，输出丢弃花的个数.
　　 每组数据后输出一个空行.

Sample Input

2
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

Sample Output

3 7
2
1 9
4
Can not put any one.

2 6
2
0 9
4
4 5
2 3

典型的线段树区间更新模板题，加一些修改就好了。

#include <iostream>
#include <stdio.h>
#include <cstring>

using namespace std;

const int N = 50000 + 5;
int sum[N<<2], lazy[N<<2], num, Min, Max; //sum数组保存当前区间花的数量

void pushUp(int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}

void pushDown(int rt,int ln,int rn)
{
    if(lazy[rt] == -1) return;
    lazy[rt<<1] = lazy[rt<<1|1] = lazy[rt];
    sum[rt<<1] = lazy[rt] * ln;
    sum[rt<<1|1] = lazy[rt] * rn;
    lazy[rt] = -1;
}

int trash(int L,int R,int l,int r,int rt)
{
    if(L<=l && r<=R){
        int t = sum[rt];
        sum[rt] = lazy[rt] = 0;
        return t;
    }
    int m = (l+r)>>1;
    pushDown(rt,m-l+1,r-m);
    int ans = 0;
    if(L <= m) ans += trash(L,R,l,m,rt<<1);
    if(m <  R) ans += trash(L,R,m+1,r,rt<<1|1);
    pushUp(rt);
    return ans;
}

void love(int L,int R,int l,int r,int rt)
{
    if(!num) return ;                       // 花插完了，直接返回即可
    if(L<=l && r<=R && sum[rt] == (r-l+1)) return ; //当前区间都已插上花，返回
    if(L<=l && r<=R && sum[rt] == 0 && num >= (r-l+1)){ //当前区间不含花，花足够即可插
        if(l < Min) Min = l;                            // 当一个区间并不为空或者不为满花时需要继续递归，缩小区间
        if(r > Max) Max = r;
        num -= (r-l+1);
        sum[rt] = (r-l+1);
        lazy[rt] = 1;
        return ;
    }
    int m = (l+r)>>1;
    pushDown(rt,m-l+1,r-m);
    if(L <= m) love(L,R,l,m,rt<<1);
    if(m <  R) love(L,R,m+1,r,rt<<1|1);
    pushUp(rt);
}

int main()
{
    int T, n, que, oper, a, b;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&que);
        memset(sum,0,sizeof(sum));
        memset(lazy,-1,sizeof(lazy));
        for(int i = 0 ; i < que ; ++i){
            scanf("%d%d%d",&oper,&a,&b);
            if(oper==1){
                num = b;
                Min = n + 1;
                Max = 0;
                love(a+1,n,1,n,1);
                if(num == b) printf("Can not put any one.\n");
                else         printf("%d %d\n", Min-1,Max-1);
            }
            else{
                printf("%d\n", trash(a+1,b+1,1,n,1));
            }
        }
        printf("\n");
    }
    return 0;
}

冲冲冲