59
分析:这道题的数据量很大,用朴素算法肯定超时,因此用线段树来做,每次操作的时间复杂度是O(log n).该题每次对某一点更新,并对一个区间的和进行查询,因此用树状数组将会比较简单,这题数据量很大,不能用cin,cout流输入输出。
树状数组解法:
#include<stdio.h> #include<iostream> #include<string.h> #define MAXN 100000 using namespace std; int c[MAXN]; int n; int lowbit(int x)///利用计算机补码计算2^k(其中k为x二进制末尾0的个数) { return x&(-x); } void add(int i,int val)///修改操作,更新C[i],如果A[i]发生改变,那么C[i]节点发生改变,C[i]的祖先也发生改变。 { while(i<=n) { c[i]+=val; i+=lowbit(i); } } int sum(int i)///计算前i项的和 { int s=0; while(i>0) { s+=c[i]; i-=lowbit(i); } return s; } int main() { int t,i,j,a,b,d; char str[13]; scanf("%d",&t); for(j=1; j<=t; j++) { memset(c,0,sizeof(c)); scanf("%d",&n); for(i=1; i<=n; i++) { scanf("%d",&a); add(i,a); } printf("Case %d:\n",j); while(~scanf("%s",str)&&strcmp(str,"End")) { scanf("%d%d",&b,&d); if(strcmp(str,"Query")==0) printf("%d\n",sum(d)-sum(b-1));///求区间的和 else if(strcmp(str,"Add")==0) add(b,d); else add(b,-d); } } }
#include<iostream> #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> #define N 50005 using namespace std; int num[N]; struct Tree { int l;///左端点 int r;///右端点 int sum;///总数 } tree[N*4]; ///总线段的长度为N,若开数组,一般开到N的4倍 void buid(int root,int l,int r)///root表示根节点,它的范围为[l,r] { tree[root].l=l; tree[root].r=r; if(tree[root].l==tree[root].r)///当左右端点相等时就是叶子节点 { tree[root].sum=num[l];///赋初值 return;///递归出口 } int mid=(l+r)/2; buid(root<<1,l,mid);///root<<1相当于root*2,即是他的左孩子 buid(root<<1|1,mid+1,r);///root<<1|1相当于root*2+1,即是他的右孩子 tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum;///父亲的sum=左孩子的sum+右孩子的sum } void update(int root,int pos,int val)///root是根节点,将在pos的点更新为val { if(tree[root].l==tree[root].r)///如果是叶子节点,即是pos对应的位置 { tree[root].sum=val;///更新操作 return;///递归出口 } int mid=(tree[root].l+tree[root].r)/2; if(pos<=mid)///如果pos点是root对应的左孩子,就调用update(root<<1,pos,val),在左孩子里找 update(root<<1,pos,val); else update(root<<1|1,pos,val); tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum; } int query(int root,int L,int R)///查询操作 { if(L<=tree[root].l&&R>=tree[root].r)///[L,R]要查询的区间包含root节点表示的区间直接返回root节点的sum值 return tree[root].sum; int mid=(tree[root].l+tree[root].r)/2,ret=0; if(L<=mid) ret+=query(root<<1,L,R);///查询root节点的左孩子 if(R>mid) ret+=query(root<<1|1,L,R);///查询root节点的右孩子 return ret;/// } int main() { int ca,cas=1,n,Q,a,b; char str[10]; scanf("%d",&ca); while(ca--) { scanf("%d",&n); for(int i=1; i<=n; i++) { scanf("%d",&num[i]);///表示在i点的兵力数量 } buid(1,1,n); printf("Case %d:\n",cas++); while(scanf("%s",str),strcmp(str,"End")) { scanf("%d%d",&a,&b); if(strcmp(str,"Query")==0) { if(a>b) swap(a,b); printf("%d\n",query(1,a,b)); } else if(strcmp(str,"Add")==0) { num[a]=num[a]+b; update(1,a,num[a]); } else if(strcmp(str,"Sub")==0) { num[a]=num[a]-b; update(1,a,num[a]); } } } }