线段树 骚操作

Vases and Flowers

-题意 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的操作，输出丢弃花的个数.
　　 每组数据后输出一个空行.  
　　 

思路：这题用线段树表示每个花瓶的状态，花瓶为空时 为 1 ，花瓶有花时 为 0 然后tree[node],表示这个区间有多少个1 也就是有多少个空花瓶。
对于操作 1 A F 我们可以查询 A 到 n中第 F 个花瓶是多少，然后找到第 F 个花瓶的位置 x ，将 A到x的区间所有元素修改为 0 也就是花瓶不为空 ，然后就是简单线段树区间修改了。
对于 操作 2 A F 也就是简单的区间查询 因为线段树维护的是 某个区间 空花瓶的个数 要想知道有花的花瓶的个数， 拿区间总共的花瓶减去空花瓶的个数。
难点 : 对于 查询 区间A 到 n 的 第 F 个空花瓶是多少 。我们可以 先查询 1到A的区间空花瓶 个数 假设空花瓶个数为cnt 然后我们只要查 区间 1到n 第cnt+F 个空花瓶是多少就可以了。 类似与权值线段树找1 到 n 的第 k大。

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+7;
int tree[4*N],add[4*N],n;
// 1表示花瓶为空 0 表示花瓶不为空
#define lson 2*node
#define rson 2*node+1
#define m (l+r)/2

void build(int l,int r,int node){
    add[node]=-1;
    if(l==r){tree[node]=1; return;}
    build(l,m,lson);
    build(m+1,r,rson);
    tree[node]=tree[lson]+tree[rson];
}

void push_down(int node,int nl,int nr){
    if(add[node]==1){
        tree[lson]=nl;
        tree[rson]=nr;
        add[lson]=add[rson]=1;
        add[node]=-1;
    }else if(add[node]==0){
        tree[lson]=0;
        tree[rson]=0;
        add[lson]=add[rson]=0;
        add[node]=-1;
    }
}

int query_c(int ql,int qr,int l,int r,int node){ // 查询ql-qr区间空花瓶的个数
    if(qr<ql)return 0;
    if(ql<=l&&qr>=r) return tree[node];
    int ans=0;
    push_down(node,m-l+1,r-m);
    if(ql<=m)ans+=query_c(ql,qr,l,m,lson);
    if(qr>m)ans+=query_c(ql,qr,m+1,r,rson);
    return ans;
}


int query(int k,int ql,int qr,int l,int r,int node){
    if(l==r)return l;
    push_down(node,m-l+1,r-m);
    if(tree[lson]>=k)return query(k,ql,qr,l,m,lson);
    else return query(k-tree[lson],ql,qr,m+1,r,rson);
}


void update(int v,int ql,int qr,int l,int r,int node){
    if(ql<=l&&qr>=r){
        add[node]=v;
        tree[node]=(r-l+1)*v;
        return;
    }
    push_down(node,m-l+1,r-m);
    if(ql<=m)update(v,ql,qr,l,m,lson);
    if(qr>m)update(v,ql,qr,m+1,r,rson);
    tree[node]=tree[lson]+tree[rson];
}

int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        int q;
        scanf("%d%d",&n,&q);
        build(1,n,1);
        while(q--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            b++;
            if(a==1){
                int cnt=query_c(b,n,1,n,1);
               if(cnt==0){
                    printf("Can not put any one.\n");
               }else if(cnt<c){
                    int back=query_c(1,b-1,1,n,1);
                    int start=query(back+1,b,n,1,n,1);
                    int end=query(back+cnt,b,n,1,n,1);
                    update(0,start,end,1,n,1);
                    printf("%d %d\n",start-1,end-1);

                }else{
                    int back=query_c(1,b-1,1,n,1);
                    int start=query(back+1,b,n,1,n,1);
                    int end=query(back+c,b,n,1,n,1);
                    update(0,start,end,1,n,1);
                    printf("%d %d\n",start-1,end-1);
                }
            }else{
                c++;
                int ans=query_c(b,c,1,n,1);
                update(1,b,c,1,n,1);
                printf("%d\n",c-b+1-ans);
            }
        }
        printf("\n");
    }
}