1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End
Case 1: 6 33 59
#include<iostream> #include<cstdio> #include<cstring> using namespace std; int n; int e[50005]; inline int lowbit(int x) { return x&(-x);//2^k } int sum(int x) //求前 N项的和 { int r=0; while(x>0) { r+=e[x]; x-=lowbit(x); } return r; } void add(int x,int d)//求前 N项sum的和 { while(x<=n) { e[x]+=d;x+=lowbit(x); } } int main() { int a,h,t,i,x,y; char str[20]; scanf("%d",&t); for(h=1;h<=t;h++) { scanf("%d",&n); memset(e,0,sizeof(e)); for(i=1;i<=n;i++) { scanf("%d",&a); add(i,a);//求前 N项sum的和 } printf("Case %d:\n",h); while(scanf("%s",str)) { if(str[0]=='E') break; scanf("%d%d",&x,&y); if(str[0]=='A') add(x,y); if(str[0]=='S') add(x,-y); if(str[0]=='Q') printf("%d\n",sum(y)-sum(x-1)); } } return 0; }
#include<stdio.h> #define maxn 500000 int c[maxn+5]; struct seg_tree { int a,b; //a为左端点 b为右端点 int total; //[a,b]区间内的人数 }s_tree[maxn*4]; //开一个 maxn*4的结构数组 //建立线段树 void build(int x,int y,int k) //在tree[k]位置建立线段[x,y] { int mid=(x+y)/2,i; s_tree[k].a=x; s_tree[k].b=y; //左右端点 s_tree[k].total=0; //初始化区间人数为0 for(i=x;i<=y;i++) s_tree[k].total+=c[i]; if(y-x>=1) { build(x,mid,k*2+1); //在左子树tree[2*k+1]上建立线段[x,mid] build(mid+1,y,k*2+2); //在右子树tree[2*k+2]上建立线段[mid+1,y] } } //查询线段树 int Query(int x,int y,int k) { //[x,y]完全覆盖结点k if(x<=s_tree[k].a&&y>=s_tree[k].b) return s_tree[k].total; int mid=(s_tree[k].a+s_tree[k].b)/2,ret; if(x<=mid) //[x,y]与左子树有交点 { //[x,y]右左子树有交点 if(y>=mid+1) ret=Query(x,y,k*2+1)+Query(x,y,k*2+2); else ret=Query(x,y,k*2+1); //[x,y]只与左子树有交点 } else ret=Query(x,y,k*2+2); //[x,y]只与右子树有交点 return ret; } //修改更新线段树 void Add(int x,int k,int p) { //x在[a,b]区间内 if(x>=s_tree[k].a&&x<=s_tree[k].b) s_tree[k].total+=p; else return; Add(x,k*2+1,p); //x是否在左子树内 Add(x,k*2+2,p); //x是否在右子树内 } int main() { int h,t,n,i; while(scanf("%d",&t)!=EOF) { for(h=1;h<=t;h++) { scanf("%d",&n); for(i=1;i<=n;i++) scanf("%d",&c[i]); build(1,n,1); char str[20]; int a,b; printf("Case %d:\n",h); while(scanf("%s",str)) { if(str[0]=='E')break; scanf("%d%d",&a,&b); if(str[0]=='A') Add(a,1,b); if(str[0]=='S') Add(a,1,-b); if(str[0]=='Q') printf("%d\n",Query(a,b,1)); } } } return 0; }
原文:http://blog.csdn.net/u012346225/article/details/42271847