HDU 1166 敌兵布阵
线段树单点更新。
题目链接http://acm.hdu.edu.cn/showproblem.php?pid=1166
模板题。我写的。
View Code
1 #include<stdio.h> 2 #include<string.h> 3 #include<stdlib.h> 4 #define N 100000 5 struct node 6 { 7 int l,r; 8 int sum; 9 }xtree[4*N]; 10 void pushup(int rn) 11 { 12 xtree[rn].sum=xtree[rn<<1].sum+xtree[rn<<1|1].sum; 13 } 14 void build(int l,int r,int rn) 15 { 16 xtree[rn].l=l; 17 xtree[rn].r=r; 18 if(l==r) 19 { 20 scanf("%d",&xtree[rn].sum); 21 return ; 22 } 23 int mid=(l+r)>>1; 24 build(l,mid,rn<<1); 25 build(mid+1,r,rn<<1|1); 26 pushup(rn); 27 } 28 void update(int num,int val,int rn) 29 { 30 if(xtree[rn].l==xtree[rn].r) 31 { 32 xtree[rn].sum+=val; 33 return ; 34 } 35 int mid=(xtree[rn].l+xtree[rn].r)>>1; 36 if(num<=mid) 37 { 38 update(num,val,rn<<1); 39 } 40 else 41 { 42 update(num,val,rn<<1|1); 43 } 44 pushup(rn); 45 } 46 int que(int le,int ri,int rn) 47 { 48 if(xtree[rn].l==le&&xtree[rn].r==ri) 49 { 50 return xtree[rn].sum; 51 } 52 int mid=(xtree[rn].l+xtree[rn].r)>>1; 53 if(le>mid) 54 { 55 return que(le,ri,rn<<1|1); 56 } 57 else if(ri<=mid) 58 { 59 return que(le,ri,rn<<1); 60 } 61 else 62 { 63 return que(le,mid,rn<<1)+que(mid+1,ri,rn<<1|1); 64 } 65 } 66 int main() 67 { 68 char str[10]; 69 int n,m,a,b,c=1; 70 scanf("%d",&n); 71 while(n--) 72 { 73 printf("Case %d:\n",c); 74 scanf("%d",&m); 75 build(1,m,1); 76 while(scanf("%s",str)) 77 { 78 if(str[0]=='E') 79 break; 80 if(str[0]=='Q') 81 { 82 scanf("%d%d",&a,&b); 83 printf("%d\n",que(a,b,1)); 84 } 85 else if(str[0]=='A') 86 { 87 scanf("%d%d",&a,&b); 88 update(a,b,1); 89 } 90 else if(str[0]=='S') 91 { 92 scanf("%d%d",&a,&b); 93 update(a,-b,1); 94 } 95 } 96 c++; 97 } 98 return 0; 99 }
底下的是HH风格的代码
View Code
1 #include <iostream> 2 #include <cstring> 3 #include <cstdio> 4 #define maxn 50007 5 using namespace std; 6 int a[maxn*4]; 7 int n; 8 void pushup(int rt) 9 { 10 a[rt] = a[rt<<1] + a[rt<<1|1]; 11 } 12 void build(int l,int r,int rt) 13 { 14 if (l == r) 15 { 16 scanf("%d",&a[rt]); 17 return ; 18 } 19 int m = (l + r)>>1; 20 build(l,m,rt<<1); 21 build(m + 1,r,rt<<1|1); 22 pushup(rt); 23 } 24 void update(int l,int r,int pos,int sc,int rt) 25 { 26 if (l == r) 27 { 28 a[rt] += sc; 29 return ; 30 } 31 int m = (l + r)>>1; 32 if (pos <= m) update(l,m,pos,sc,rt<<1); 33 else update(m + 1,r,pos,sc,rt<<1|1); 34 pushup(rt); 35 } 36 int query(int l,int r,int L,int R,int rt) 37 { 38 if (l >= L && r <= R) 39 { 40 return a[rt]; 41 } 42 int m = (l + r)>>1; 43 int res = 0; 44 if (L <= m) res += query(l,m,L,R,rt<<1); 45 if (R > m) res += query(m + 1,r,L,R,rt<<1|1); 46 return res; 47 } 48 int main() 49 { 50 //freopen("in.txt","r",stdin); 51 int t,b,c,cas = 1; 52 char op[10]; 53 scanf("%d",&t); 54 while (t--) 55 { 56 printf("Case %d:\n",cas++); 57 scanf("%d",&n); 58 build(0,n - 1,1); 59 while (scanf("%s",op)) 60 { 61 if (op[0] == 'E') break; 62 scanf("%d%d",&b,&c); 63 if (op[0] == 'A') 64 { 65 update(0,n - 1,b - 1,c,1); 66 } 67 else if (op[0] == 'S') 68 { 69 update(0,n - 1,b - 1,-c,1); 70 } 71 else 72 { 73 int ans = query(0,n - 1,b - 1,c - 1,1); 74 printf("%d\n",ans); 75 } 76 } 77 } 78 return 0; 79 }
