Codevs-4919 线段树练习4(区间加上一个值并求摸个区间整除k的数的个数，线段树+数组维护)

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<cmath>
  5 #include<cstdlib>
  6 #include<algorithm>
  7 #include<queue>
  8 #include<stack>
  9 #include<set>
 10 #include<vector>
 11 #include<map>
 12 using namespace std;
 13 typedef long long LL;
 14 const int INF=0x3f3f3f3f;
 15 const LL inf=0x3f3f3f3f3f3f3f3fLL;
 16 const int maxn=1e5+10;
 17 int n,q,c[maxn],a,b,x,temp[7];
 18 string str;
 19 
 20 struct Node{
 21     int l,r,tag,mod[7];
 22 } tree[maxn<<2];
 23 
 24 void build(int pos,int l,int r)
 25 {
 26     tree[pos].l=l,tree[pos].r=r,tree[pos].tag=0;
 27     for(int i=0;i<7;i++) tree[pos].mod[i]=0;
 28     if(l==r)
 29     {
 30         tree[pos].mod[c[l]%7]=1;
 31         return ;
 32     }
 33     int mid=(tree[pos].l+tree[pos].r)>>1;
 34     build(pos<<1,l,mid);
 35     build(pos<<1|1,mid+1,r);
 36     for(int i=0;i<7;i++) tree[pos].mod[i]=tree[pos<<1].mod[i]+tree[pos<<1|1].mod[i];
 37 }
 38 
 39 void change(int pos,int x)
 40 {
 41     for(int i=0;i<7;i++) temp[(i+x)%7]=tree[pos].mod[i];
 42     for(int i=0;i<7;i++) tree[pos].mod[i]=temp[i];
 43 }
 44 
 45 void pushup(int pos)
 46 {
 47     for(int i=0;i<7;i++) 
 48         tree[pos].mod[i]=tree[pos<<1].mod[i]+tree[pos<<1|1].mod[i];
 49 }
 50 
 51 void pushdown(int pos)
 52 {
 53     if(tree[pos].l==tree[pos].r) return ;
 54     tree[pos<<1].tag+=tree[pos].tag;
 55     tree[pos<<1|1].tag+=tree[pos].tag;
 56     change(pos<<1,tree[pos].tag),change(pos<<1|1,tree[pos].tag);
 57     tree[pos].tag=0;
 58 }
 59 
 60 void update(int pos,int l,int r,int x)
 61 {
 62     if(tree[pos].tag) pushdown(pos);
 63     if(tree[pos].l==l&&tree[pos].r==r)
 64     {
 65         tree[pos].tag+=x;change(pos,x);
 66         return ;
 67     }
 68     int mid=(tree[pos].l+tree[pos].r)>>1;
 69     if(r<=mid) update(pos<<1,l,r,x);
 70     else if(l>=mid+1) update(pos<<1|1,l,r,x);
 71     else update(pos<<1,l,mid,x),update(pos<<1|1,mid+1,r,x);
 72     pushup(pos);
 73 }
 74 
 75 int query(int pos,int l,int r)
 76 {
 77     if(tree[pos].tag) pushdown(pos);
 78     if(tree[pos].l==l&&tree[pos].r==r) return tree[pos].mod[0];
 79     int mid=(tree[pos].l+tree[pos].r)>>1;
 80     if(r<=mid) return query(pos<<1,l,r);
 81     else if(l>=mid+1) return query(pos<<1|1,l,r);
 82     else return query(pos<<1,l,mid)+query(pos<<1|1,mid+1,r);    
 83 }
 84 
 85 int main()
 86 {
 87     ios::sync_with_stdio(false);
 88     cin>>n;
 89     for(int i=1;i<=n;i++) cin>>c[i];
 90     build(1,1,n);
 91     cin>>q;
 92     while(q--)
 93     {
 94         cin>>str;
 95         if(str[0]==‘a‘)
 96         {
 97             cin>>a>>b>>x;
 98             update(1,a,b,x);
 99         }
100         else if(str[0]==‘c‘)
101         {
102             cin>>a>>b;
103             cout<<query(1,a,b)<<endl;
104         }
105     }
106     
107     return 0;
108 }