codevs 4919 線段樹練習4
阿新 • • 發佈:2017-06-26
sum struct () count query const == add pro
題目描述 Description
給你N個數,有兩種操作
1:給區間[a,b]內的所有數都增加X
2:詢問區間[a,b]能被7整除的個數
輸入描述 Input Description第一行一個正整數n,接下來n行n個整數,再接下來一個正整數Q,表示操作的個數. 接下來Q行每行若幹個整數。如果第一個數是add,後接3個正整數a,b,X,表示在區間[a,b]內每個數增加X,如果是count,表示統計區間[a,b]能被7整除的個數
輸出描述 Output Description對於每個詢問輸出一行一個答案
樣例輸入 Sample Input
3 2 3 4 6 count 1 3 count 1 2 add 1 3 2 count 1 3 add 1 3 3 count 1 3
樣例輸出 Sample Output
0
0
0
1
數據範圍及提示 Data Size & Hint10%:1<N<=10,1<Q<=10
30%:1<N<=10000,1<Q<=10000
100%:1<N<=100000,1<Q<=100000
//每個節點用一個數組來維護 %7的 = 0 ,1 , 2 , 3 , 4 , 5 , 6 的個數 #include<iostream> #include<cstdio> #include<cstring> #include<string> #include<cmath> #include<algorithm> using namespace std; const int N=100001; int n,x,y,z,ans,Q; int tmp[7]; string s; struct node { int l,r; int f; int sum[7]; }T[N*4]; inline int read() { int x=0; char c=getchar(); while(c<‘0‘||c>‘9‘)c=getchar(); while(c>=‘0‘&&c<=‘9‘)x=x*10+c-‘0‘,c=getchar(); return x; } void up(int jd) { for(int i=0;i<7;i++) T[jd].sum[i]=T[jd<<1].sum[i]+T[jd<<1|1].sum[i]; } void build(int jd,int l,int r) { T[jd].l=l; T[jd].r=r; if(l==r) { scanf("%d",&x); x=read(); T[jd].sum[x%7]++; return ; } int mid=l+r>>1; build(jd<<1,l,mid); build(jd<<1|1,mid+1,r); up(jd); } void down(int jd) { for(int i=0;i<7;i++) tmp[i]=T[jd<<1].sum[i]; for(int i=0;i<7;i++) T[jd<<1].sum[(i+T[jd].f)%7]=tmp[i]; for(int i=0;i<7;i++) tmp[i]=T[jd<<1|1].sum[i]; for(int i=0;i<7;i++) T[jd<<1|1].sum[(i+T[jd].f)%7]=tmp[i]; T[jd<<1].f+=T[jd].f; T[jd<<1|1].f+=T[jd].f; T[jd].f=0; } void add(int jd) { if(T[jd].l>=x&&T[jd].r<=y) { for(int i=0;i<7;i++) tmp[i]=T[jd].sum[i]; for(int i=0;i<7;i++) T[jd].sum[(i+z)%7]=tmp[i]; T[jd].f+=z; return; } if(T[jd].f) down(jd); int mid=T[jd].l+T[jd].r>>1; if(x<=mid) add(jd<<1); if(y>mid) add(jd<<1|1); up(jd); } void query(int jd) { if(T[jd].l>=x&&T[jd].r<=y) { ans+=T[jd].sum[0]; return; } if(T[jd].f) down(jd); int mid=T[jd].l+T[jd].r>>1; if(x<=mid) query(jd<<1); if(y>mid) query(jd<<1|1); } int main() { n=read(); build(1,1,n); Q=read(); while(Q--) { cin>>s; if(s[0]==‘a‘) { x=read();y=read();z=read(); z%=7; add(1); } else { x=read();y=read(); ans=0; query(1); printf("%d\n",ans); } } return 0; }
codevs 4919 線段樹練習4