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 <algorithm> #include <cstdio> #include <cstring> using namespace std; #define M 50005 int a[M]; struct Node{ int left; int right; int count; }node[4*M]; //开4倍大是因为变成二叉树后节点有很多。 void TreeMake(int l,int r,int i) //建立二叉树。 { node[i].left=l; //l是区间前界。 node[i].right=r; //r是区间后界。 if(l==r) //区间长度只有1,那就可以得到确切的值。 { node[i].count=a[l]; return; } int m=(l+r)/2; //否则还要接着划分区间。 TreeMake(l,m,2*i); TreeMake(m+1,r,2*i+1); node[i].count=node[2*i].count+node[2*i+1].count; //得到这个区间的和。 } void TreeUpdate(int i,int x,int op,int num) //更新。 { int l=node[i].left; int r=node[i].right; int mid=(l+r)/2; if(x==l && x==r) //找到那个要更新的节点,就更新它。 { if(op==1) node[i].count+=num; else node[i].count-=num; return; } if(x>mid) //判断那个要更新的点是在左子树还是在右子树。 TreeUpdate(2*i+1,x,op,num); //进入左子树。 else TreeUpdate(2*i,x,op,num); //进入右子树。 if(op==1) node[i].count+=num; //这里就是对每包含该节点的大区间都更新和值。 else node[i].count-=num; } int TreeQuery(int l,int r,int i) //寻找要输出和值的区间。 { int m=(node[i].left+node[i].right)/2; if(node[i].left==l && node[i].right==r) return node[i].count; // if(node[i].left==node[i].right) return 0; //发现没有这句也能过,不知道是什么,如果知道这句话是干什么的,请告知。 if(r<=m) //如果后界比中间值小,整个区间都在左子树内。 return TreeQuery(l,r,i*2); else if(l>m) //同理,如果前界比中间值大,整个区间都在右子树内。 return TreeQuery(l,r,2*i+1); else //如果区间被中间值分开,就分别求出来,再相加。 return TreeQuery(l,m,2*i)+TreeQuery(m+1,r,2*i+1); } int main() { int m,i,j,n,tmp; char cmd[10]; int x,y; scanf("%d",&m); for(i=1;i<=m;i++) { printf("Case %d:\n",i); scanf("%d",&n); for(j=1;j<=n;j++) { scanf("%d",&a[j]); } TreeMake(1,n,1); while(scanf("%s",cmd)) { if (cmd[0] == ‘E‘) break; scanf("%d %d", &x, &y); if (cmd[0] == ‘Q‘) { tmp = TreeQuery(x, y, 1); printf("%d\n", tmp); } else if (cmd[0] == ‘A‘) { TreeUpdate(1, x, 1, y); } else { TreeUpdate(1, x, 2, y); } } } return 0; }
HDU 1166 敌兵布阵(线段树),布布扣,bubuko.com
原文地址:http://blog.csdn.net/qq2256420822/article/details/38391195