CC Sereja and Ballons (主席树)

转载请注明出处，谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents    by---cxlove

题意：有n个盒子，每个盒子有若干个气球，每次操作可以拿走某个盒子的一个气球，然后 给出一些区间，问每次操作后有多少个区间的盒子全为空。

http://www.codechef.com/AUG13/problems/SEABAL

做法：用链表维护一下当前位置前一个非0位置，下一个非0位置。

如果当前位置为k，前一个非0位置为l，下一个非0位置为r。

那么当前位置减为了0，此时增加的区间应该是左端点在（l，k] ,右端点在[k，r）。

那么接下来就是查询一下有多少个区间满足这个条件。

用主席树维护以每个位置为区间右端点的左端点位置。

#include <iostream>#include <cstdio>#include <set>#include <vector>#include <cstring>#include <algorithm>#define mem(a,b) memset (a , b , sizeof(a))using namespace std;const int N = 100005;const int M = 30000005;struct Node {    int l , r;    void input () {        scanf ("%d %d" , &l , &r);    }    bool operator < (const Node &n) const {        return r < n.r;    }}que[N];int n , m , q , a[N]; int T[M] , lson[M] , rson[M] , sum[M] , tot = 0;int l[N] , r[N];int bulid (int l , int r) {    int root = ++ tot;    sum[root] = 0;    if (l != r) {        int m = (l + r) >> 1;        lson[root] = bulid (l , m);        rson[root] = bulid (m + 1 , r);    }    return root;}int update (int root , int l , int r , int pos) {    int newroot = ++ tot;    sum[newroot] = sum[root] + 1;    if (l != r) {        int m = (l + r) >> 1;        if (pos <= m) {            lson[newroot] = update (lson[root] , l , m , pos);            rson[newroot] = rson[root];        }        else {            rson[newroot] = update (rson[root] , m + 1 , r , pos);            lson[newroot] = lson[root];        }    }    //cout << sum[newroot] << " " << sum[lson[newroot]] << " " << sum[rson[newroot]] << endl;;    return newroot;}int query (int root , int L , int R , int l , int r) {    if (L == l && R == r) {        return sum[root];    }    int m = (L + R) >> 1;    if (r <= m) return query (lson[root] , L , m , l , r);    else if (l > m) return query (rson[root] , m + 1 , R , l , r);    else return query (lson[root] , L , m , l , m) + query (rson[root] , m + 1 , R , m + 1 , r);}void out (int root , int l , int r) {    cout << l << " " << r << " " << sum[root] << endl;    if (l != r) {        int m = (l + r) >> 1;        out (lson[root] , l , m);        out (rson[root] , m + 1 , r);    }}int main () {    // freopen ("input.txt" , "r" , stdin);    // freopen ("output.txt" , "w" , stdout);    scanf ("%d %d" , &n , &m);    l[0] = 0; r[n + 1] = n + 1;    for (int i = 1 ; i <= n ; i ++) {        scanf ("%d" , &a[i]);        l[i] = i - 1;        r[i] = i + 1;    }    for (int i = 0 ; i < m ; i ++) {        que[i].input();    }    sort (que , que + m);    T[0] = bulid (1 , n);    for (int i = 1 , j = 0 ; i <= n ; i ++) {        T[i] = T[i - 1];        while (j < m && que[j].r == i) {            T[i] = update (T[i] , 1 , n , que[j].l);            j ++;        }        // cout << "root : " << i << endl;        // out (T[i] , 1 , n);    }    scanf ("%d" , &q);    int ans = 0;    while (q --) {        int k;        scanf ("%d" , &k);        k += ans;        //cout << k << endl;        a[k] --;        if (a[k] == 0) {            int L = l[k] , R = r[k];            int ret = query (T[R - 1] , 1 , n , L + 1 , k) - query (T[k - 1] , 1 , n , L + 1 , k);             ans += ret;            // cout << "ret :" << ret << endl;            if (r[k] <= n) l[r[k]] = l[k];              if (l[k] >= 1) r[l[k]] = r[k];        }        printf ("%d\n" , ans);                    }    return 0;}