标签:
Description
Input
Output
Sample Input
Sample Output
#include <stdio.h> #include <string.h> #include <algorithm> using namespace std; #define N 100010<<2//一棵树的某排有N个数,那此排以前就一共有N-1个数,且此排的下一排有2N个数,N+N-1+2N约等于4N,要开N的四倍数组,才不会越界 int sto[N],up[N],x,y; char s[3]; 8 void init(int t,int w,int o)//初始化树中的某一结点,t为该结点左边界,w为右边界,o为该结点的标号(用来表示该结点的位置) { up[o]=0;//up[o]代表该结点的值需要改变时,增加或减少的量; if(t==w)//当递归到最底层,即划分到区间长度为0时 { scanf("%d",&sto[o]);//输出该结点的数值 return; } int mid=(t+w)>>1;//除以2,划分下一层的边界 int lso=o<<1;//乘以二,求出其左子数的标号 init(t,mid,lso);//递归到其左子树 init(mid+1,w,lso|1); sto[o]=sto[lso]+sto[lso|1];//该结点的左子树的值加上右子树的值,从下到上进行累加,获得整棵数的值得累加和 } //某个结点代表的是某个范围,此结点想要变化N,就代表这一范围的每个数都要变化N; void updown(int sum,int o)//更新某个结点的值的操作,sum指该结点以下有多少个结点(他们同样要一起更新),o代表该结点的标号 {//第一次修改值得时候,是不会用到的,因为会有多组测试数据,会进行多次修改,所以才要更新 if(up[o])//如果变化量不为0,则需要进行更新 { int lso=o<<1;//此结点的左子树标号 up[lso]+=up[o];//把变化量传给左子数 up[lso|1]+=up[o];//把变化量传给右子数 sto[lso]+=(sum-(sum>>1))*up[o];//左子树的值为它自身加上它以下的所有结点数变化量之和(左子树的值要多1,所以要加多的那一边) sto[lso|1]+=(sum>>1)*up[o]; up[o]=0;//用完了就制为0,以免后面重复加 } } int updata(int t,int w,int o,int st,int en,int num)//查找并计算出某颗树的值,t,w为整棵树的范围,o,代表查找的位置,st,en,代表该位置的范围,num代表变化量 { if(st<=t&&w<=en)//当查到那个那个位置的范围以内时 { if(num!=1<<29)//如果num不为无穷大,即除了查询操作,还要进行更新工作 { up[o]+=num;//把要加的变量付给up[o] sto[o]+=(w-t+1)*num;//算出该范围总的变化量 } return sto[o];//返回求和 } updown(w-t+1,o);//继续向下更新 int mid=(t+w)>>1;//继续缩小范围 int lso=o<<1; int ans=0; if(st<=mid) ans+=updata(t,mid,lso,st,en,num);//当st小于mid 则表明该范围至少有一部分在左子树;则减小范围,再次进行查找计算,然后一层一层退回; if(en>mid) ans+=updata(mid+1,w,lso|1,st,en,num);//当en>mid 说明,该范围至少有一部分在右子树,则减小范围,再次进行查找计算,然后层层退回; sto[o]=sto[lso]+sto[lso|1];//求出每一个范围的总值; return ans;//每次返回回去都不再重新走ans=0,ans一直累加,直到输出答案 } int main() { int n,m,t=1; scanf("%d",&n); while(n--) { scanf("%d",&m); init(1,m,1); printf("Case %d:\n",t++); while(1) { scanf("%s",s); if(s[0]==‘Q‘) { scanf("%d%d",&x,&y); printf("%d\n",updata(1,m,1,x,y,1<<29));//把查询范围x到y的和为多少 } if(s[0]==‘A‘) { scanf("%d%d",&x,&y); updata(1,m,1,x,x,y);//把范围订在自己到自己 } if(s[0]==‘S‘) { scanf("%d%d",&x,&y); updata(1,m,1,x,x,-y); } if(s[0]==‘E‘) break; } } return 0; }
标签:
原文地址:http://www.cnblogs.com/tianmin123/p/4757953.html