CodeVS4919 线段树练习4
题目链接：http://codevs.cn/problem/4919/
题目描述 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
对于每个询问输出一行一个答案
一道水题。把区间内除以7余0,1,2,3,4,5,6的数的个数都记录下来，更新时维护一下就好了。
1 #include <iostream> 2 #include <cstdio> 3 #include <algorithm> 4 #include <cstring> 5 #define rep(i,l,r) for(int i = l; i <= r; i++) 6 #define clr(x,y) memset(x,y,sizeof(x)) 7 using namespace std; 8 const int INF = 0x3f3f3f3f; 9 const int maxn = 100010; 10 inline int read(){ 11 int ans = 0, f = 1; char c = getchar(); 12 for(; !isdigit(c); c = getchar()) if (c == '-') f = -1; 13 for(; isdigit(c); c = getchar()) ans = ans * 10 + c - '0'; 14 return ans * f; 15 } 16 struct Node{ 17 int l,r,mod[7],f; 18 }t[maxn<<2]; 19 int s[maxn],tmp[7],a,b,x,n,m; 20 char c[10]; 21 inline void change(int w,int p){ 22 rep(i,0,6) tmp[(i+p)%7] = t[w].mod[i]; 23 rep(i,0,6) t[w].mod[i] = tmp[i]; 24 } 25 inline void maintain(int w){ 26 rep(i,0,6) t[w].mod[i] = t[w<<1].mod[i] + t[w<<1|1].mod[i]; 27 } 28 inline void pushdown(int w){ 29 if (t[w].l == t[w].r || (!t[w].f)) return; 30 t[w<<1].f += t[w].f; t[w<<1|1].f += t[w].f; 31 change(w<<1,t[w].f); change(w<<1|1,t[w].f); t[w].f = 0; 32 } 33 void build(int u,int v,int w){ 34 t[w].l = u; t[w].r = v; t[w].f = 0; 35 rep(i,0,6) t[w].mod[i] = 0; 36 if (u == v){ 37 t[w].mod[s[u]%7] = 1; 38 return; 39 } 40 int mid = (u + v) >> 1; 41 build(u,mid,w<<1); build(mid+1,v,w<<1|1); 42 maintain(w); 43 } 44 void modify(int u,int v,int w){ 45 pushdown(w); 46 if (t[w].l == u && t[w].r == v){ 47 t[w].f += x; change(w,x); return; 48 } 49 int mid = (t[w].l + t[w].r) >> 1; 50 if (v <= mid) modify(u,v,w<<1); 51 else if (u > mid) modify(u,v,w<<1|1); 52 else{ 53 modify(u,mid,w<<1); 54 modify(mid+1,v,w<<1|1); 55 } 56 maintain(w); 57 } 58 int query(int u,int v,int w){ 59 pushdown(w); 60 if (t[w].l == u && t[w].r == v) return t[w].mod[0]; 61 int mid = (t[w].l + t[w].r) >> 1; 62 if (v <= mid) return query(u,v,w<<1); 63 else if (u > mid) return query(u,v,w<<1|1); 64 else return query(u,mid,w<<1) + query(mid+1,v,w<<1|1); 65 } 66 int main(){ 67 n = read(); rep(i,1,n) s[i] = read(); 68 build(1,n,1); 69 m = read(); rep(i,1,m){ 70 scanf("%s",c); 71 if (c[0] == 'a'){ 72 a = read(); b = read(); x = read(); 73 modify(a,b,1); 74 } 75 else{ 76 a = read(); b = read(); 77 printf("%d\n",query(a,b,1)); 78 } 79 } 80 return 0; 81 }
