【北京集训测试赛(七)/BZOJ4631】踩气球(Balloon)-线段树-STL-Vector-CSDN博客

Problem 踩气球

题目大意

给m个区间和一个数列，每次对其中一个数减一，强制在线求每次操作后区间内数字全都为0的区间。

Solution

对于数列开一个线段树。对于线段树每一个点开一个vector存区间的id。

这样我们一共最多会存下$\log m$个id。

对于每次操作，维护线段树sum值。

当我们将一个数减到了零，那么对于到根路径上的每一个点扫一遍。

如果这个点的sum值变成零了那么就扫一遍这个点的vector

判断这个id的区间是否被减完了（额外开一个数组维护就行了）

如果减成零了那么就ans++。最后我们清空这个vector。

这样下来，因为每个vector中的东西最多只会访问一遍(然后就删了)，

所以复杂度是$O(q \log n+\log m)$

AC Code

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <vector>
 5 using namespace std;
 6 struct Node{
 7     int l,r,sum;
 8     vector<int> v;
 9 }a[500010];
10 int lastans=0,bal[100010],ans=0,x,f[100010];
11 int q,l,r,m,n;
12 void sg_update(int now){
13     if(a[now].sum)return;
14     for(vector<int>::iterator it=a[now].v.begin();it!=a[now].v.end();it++){
15         bal[*it]-=a[now].r-a[now].l+1;
16         if(!bal[*it])ans++;
17     }
18     a[now].v.clear();
19 }
20 void sg_build(int l,int r,int now){
21     a[now].l=l;a[now].r=r;
22     if(l==r){
23         a[now].sum=f[l];
24         return;
25     }
26     int mid=(l+r)>>1;
27     sg_build(l,mid,now<<1);
28     sg_build(mid+1,r,now<<1|1);
29     a[now].sum=a[now<<1].sum+a[now<<1|1].sum;
30     return;
31 }
32 int sg_delete(int now,int x){
33     if(a[now].l==a[now].r){
34         a[now].sum--;
35         sg_update(now);
36         return ans;
37     }
38     int mid=(a[now].l+a[now].r)>>1;
39     if(x<=mid)sg_delete(now<<1,x);
40     else sg_delete(now<<1|1,x);
41     a[now].sum--;
42     sg_update(now);
43     return ans;
44 }
45 void sg_add(int l,int r,int now,int x){
46     if(l<=a[now].l&&r>=a[now].r){
47         a[now].v.push_back(x);
48         bal[x]=r-l+1;
49         return;
50     }
51     int mid=(a[now].l+a[now].r)>>1;
52     if(l<=mid)sg_add(l,r,now<<1,x);
53     if(mid<r)sg_add(l,r,now<<1|1,x);
54 
55 }
56 int main(){
57 //  freopen("c.in","r",stdin);
58     scanf("%d%d",&n,&m);
59     for(int i=1;i<=n;i++)
60         scanf("%d",&f[i]);
61     sg_build(1,n,1);
62     for(int i=1;i<=m;i++)
63         scanf("%d%d",&l,&r),
64         sg_add(l,r,1,i);
65     scanf("%d",&q);
66     for(int i=1;i<=q;i++)
67         scanf("%d",&x),
68         lastans=sg_delete(1,(x+lastans-1)%n+1),
69         printf("%d\n",ans);
70     return 0;
71 }