POJ 3468 A Simple Problem With Integers

 

线段树的应用，对区间内的值进行加减，然后求区间内的数字和

一开始，对数据进行加减时，更新到每个点，超时了。。。

然后按照网上说的设置了增量值d，当更新区间与当前节点的区间重合时，更新d值，然后向上更新sum值

当查询区间和时，将父区间的增量和传到当前区间

 

还有，函数的参数类型要和传入参数的类型相一致，因为这个WA一次。。。

代码：

#include<iostream> using namespace std; struct node { int left; int right; __int64 sum; __int64 d;//表示增量 }t[500001]; __int64 a[100005]; char com[5]; void build(int l,int r,int n) { t[n].left=l; t[n].right=r; if(l==r) { t[n].sum=a[l]; return; } int mid=(l+r)>>1; build(l,mid,n*2); build(mid+1,r,n*2+1); t[n].sum=t[n*2].sum+t[n*2+1].sum; } void update(int l,int r,int n,__int64 v) { if(t[n].left==l&&t[n].right==r)//更新当前区间的增量值 { t[n].d+=v; return; } int mid=(t[n].left+t[n].right)>>1; if(mid>=r) update(l,r,n*2,v); else if(mid<l) update(l,r,n*2+1,v); else { update(l,mid,n*2,v); update(mid+1,r,n*2+1,v); } t[n].sum=t[n*2].sum+(t[n*2].right-t[n*2].left+1)*t[n*2].d;//更新父区间的和 t[n].sum+=t[n*2+1].sum+(t[n*2+1].right-t[n*2+1].left+1)*t[n*2+1].d; } __int64 query(int l,int r,int n,__int64 p)//p为由父区间传递下来的子区间的增量的改变量 { p+=t[n].d; if(t[n].left==l&&t[n].right==r) { return t[n].sum+(t[n].right-t[n].left+1)*p; } int mid=(t[n].left+t[n].right)>>1; if(mid>=r) { return query(l,r,n*2,p); } else if(mid<l) { return query(l,r,n*2+1,p); } else { __int64 s1=query(l,mid,n*2,p); __int64 s2=query(mid+1,r,n*2+1,p); return s1+s2; } } int main() { int i,j,q,n; __int64 v,sum; scanf("%d %d",&n,&q); for(i=1;i<=n;i++) scanf("%I64d",&a[i]); build(1,n,1); while(q--) { scanf("%s",&com); if(com[0]=='Q') { scanf("%d %d",&i,&j); sum=query(i,j,1,0); printf("%I64d/n",sum); } else { scanf("%d %d %I64d",&i,&j,&v); update(i,j,1,v); } } system("pause"); return 0; }