转载请注明出处,谢谢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;
}