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;

}