HDU - 1166 - 敌兵布阵 线段树的单点修改,区间求和
#include<iostream> #include<stdio.h> #include<string.h> #include<algorithm> using namespace std; const int maxx = 50030; int arry[maxx]; struct node { int left; int right; int mid; int val; } a[maxx*4]; void pushup(int root) { a[root].val=a[root<<1].val+a[root<<1|1].val;//把laze标记往上传递。不能是加等于,就是等于,因为底部被更新后,需要往上的是新的内容 } void buildtree(int root,int l,int r) { int mid=(l+r)/2; a[root].left=l; a[root].right=r; a[root].mid=mid; a[root].val=0; if(l==r) { a[root].val=arry[l];//到最低端的时候 return; } buildtree(root*2,l,mid); buildtree(root*2+1,mid+1,r); pushup(root);//把这个区间创建完成后,需要把更新往上更新 } void updata(int root,int tag,int val)//单点更新 { int l = a[root].left,r=a[root].right; if(l==r) { a[root].val+=val; return; } int mid=(l+r)/2; if(tag<=mid)updata(2*root,tag,val); else updata(root*2+1,tag,val); pushup(root); } int query(int root,int ql,int qr) { int res=0; int l=a[root].left,r=a[root].right; int mid=(l+r)/2; if(ql<=l && qr>=r)return a[root].val;//询问的区间是比这个区间大的,那么代表这个区间的内部的所有都要被选到,因此无需再往下遍历 if(ql<=mid)res+=query(root*2,ql,qr);//询问的左区间比当前区间小,那么这个区间的左边有一部分取不到,因此需要继续往下找 if(qr>mid)res+=query(root*2+1,ql,qr);//询问的右区间比当前的右区间大,那么这个区间的右边有一部分取不到,因此也需要往下找 return res;//最后把这两边的内容传递回去即可 } int main() { int t; int n; scanf("%d",&t); char op[20]; int pos,val; int l,r; int cas=0; while(t--) { scanf("%d",&n); printf("Case %d:\n",++cas); for (int i=1; i<=n; i++) { scanf("%d",&arry[i]); } buildtree(1,1,n); while(~scanf("%s",op)) { if(op[0]=='E')break; if(op[0]=='A') { scanf("%d%d",&pos,&val); updata(1,pos,val); } else if(op[0]=='S') { scanf("%d%d",&pos,&val); // cout<<"???"<<-val<<endl; updata(1,pos,-val); } else { scanf("%d%d",&l,&r); printf("%d\n", query(1,l,r)); } } } return 0; }
有不懂欢迎咨询
QQ:1326487164(添加时记得备注)