hdoj 1166 敌兵布阵

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1166

解题思路:线段树(单点更新)

  1 /**************************************************************************
  2 user_id: SCNU20102200088
  3 problem_id: hdoj 1166
  4 problem_name: 敌兵布阵
  5 **************************************************************************/
  6 
  7 #include <algorithm>
  8 #include <iostream>
  9 #include <iterator>
 10 #include <iomanip>
 11 #include <sstream>
 12 #include <fstream>
 13 #include <cstring>
 14 #include <cstdlib>
 15 #include <climits>
 16 #include <bitset>
 17 #include <string>
 18 #include <vector>
 19 #include <cstdio>
 20 #include <cctype>
 21 #include <ctime>
 22 #include <cmath>
 23 #include <queue>
 24 #include <stack>
 25 #include <list>
 26 #include <set>
 27 #include <map>
 28 using namespace std;
 29 
 30 //线段树
 31 #define lson l,m,rt<<1
 32 #define rson m+1,r,rt<<1|1
 33 
 34 //手工扩展栈
 35 #pragma comment(linker,"/STACK:102400000,102400000")
 36 
 37 const double EPS=1e-9;
 38 const double PI=acos(-1.0);
 39 const double E=2.7182818284590452353602874713526;  //自然对数底数
 40 const double R=0.5772156649015328606065120900824;  //欧拉常数:(1+1/2+...+1/n)-ln(n)
 41 
 42 const int x4[]={-1,0,1,0};
 43 const int y4[]={0,1,0,-1};
 44 const int x8[]={-1,-1,0,1,1,1,0,-1};
 45 const int y8[]={0,1,1,1,0,-1,-1,-1};
 46 
 47 typedef long long LL;
 48 
 49 typedef int T;
 50 T max(T a,T b){ return a>b? a:b; }
 51 T min(T a,T b){ return a<b? a:b; }
 52 T gcd(T a,T b){ return b==0? a:gcd(b,a%b); }
 53 T lcm(T a,T b){ return a/gcd(a,b)*b; }
 54 
 55 ///////////////////////////////////////////////////////////////////////////
 56 //Add Code:
 57 const int maxn=50005;
 58 int sum[maxn<<2];
 59 
 60 void pushup(int rt){
 61     sum[rt]=sum[rt<<1]+sum[rt<<1|1];
 62 }
 63 
 64 void build(int l,int r,int rt){
 65     if(l==r){
 66         scanf("%d",&sum[rt]);
 67         return ;
 68     }
 69     int m=(l+r)>>1;
 70     build(lson);
 71     build(rson);
 72     pushup(rt);
 73 }
 74 
 75 void update(int p,int add,int l,int r,int rt){
 76     if(l==r){
 77         sum[rt]+=add;
 78         return ;
 79     }
 80     int m=(l+r)>>1;
 81     if(p<=m) update(p,add,lson);
 82     else update(p,add,rson);
 83     pushup(rt);
 84 }
 85 
 86 int query(int L,int R,int l,int r,int rt){
 87     if(L<=l && R>=r) return sum[rt];
 88     int m=(l+r)>>1,ret=0;
 89     if(L<=m) ret+=query(L,R,lson);
 90     if(R>m) ret+=query(L,R,rson);
 91     return ret;
 92 }
 93 ///////////////////////////////////////////////////////////////////////////
 94 
 95 int main(){
 96     std::ios::sync_with_stdio(false);
 97     //freopen("in.txt","r",stdin);
 98     //freopen("out.txt","w",stdout);
 99     ///////////////////////////////////////////////////////////////////////
100     //Add Code:
101     int Case,n,i,a,b;
102     char c[10];
103     scanf("%d",&Case);
104     for(i=1;i<=Case;i++){
105         scanf("%d",&n);
106         build(1,n,1);
107         printf("Case %d:\n",i);
108         while(scanf("%s",c),c[0]!='E'){
109             scanf("%d%d",&a,&b);
110             if(c[0]=='A') update(a,b,1,n,1);
111             else if(c[0]=='S') update(a,-b,1,n,1);
112             else printf("%d\n",query(a,b,1,n,1));
113         }
114     }
115     ///////////////////////////////////////////////////////////////////////
116     return 0;
117 }
118 
119 /**************************************************************************
120 Testcase:
121 Input:
122 1
123 10
124 1 2 3 4 5 6 7 8 9 10
125 Query 1 3
126 Add 3 6
127 Query 2 7
128 Sub 10 2
129 Add 6 3
130 Query 3 10
131 End
132 Output:
133 Case 1:
134 6
135 33
136 59
137 **************************************************************************/

posted on 2013-08-16 00:29  SCNU20102200088  阅读(140)  评论(0编辑  收藏  举报

导航