poj3468_线段树的应用

题意：给定一个数组，给定两种区间操作，一种是往一个区间中的每一个位置加上一个数，另一种是询问某个区间的数的和

解法：线段树

结构定义：

struct Node{int l,r;long long tsum,csum; //其中tsum表示对整个区间的改变量，例如加入[1,5]，则所有在区间[1,5]的祖先位置的都要加//csum表示该区间的改变Node(){l=r=tsum=csum=0;}};

代码如下：

#include <iostream>using namespace std;#define MAXN 100005struct Node{int l,r;long long tsum,csum; //其中tsum表示对整个区间的改变量，例如加入[1,5]，则所有在区间[1,5]的祖先位置的都要加//csum表示刚好该区间的改变Node(){l=r=tsum=csum=0;}};Node treeNode[4*MAXN];void construct(int p,int l,int r){treeNode[p].l=l;treeNode[p].r=r;if (l==r){return;}int mid=(treeNode[p].l+treeNode[p].r)/2;construct(2*p,l,mid);construct(2*p+1,mid+1,r);}void Addnode(int p,int l,int r,int val){treeNode[p].tsum+=(r-l+1)*val;if (treeNode[p].l==l&&treeNode[p].r==r){treeNode[p].csum+=val;return;}int mid=(treeNode[p].l+treeNode[p].r)/2;if (r<=mid){Addnode(2*p,l,r,val);}else if (l>=mid+1){Addnode(2*p+1,l,r,val);}else{Addnode(2*p,l,mid,val);Addnode(2*p+1,mid+1,r,val);}}long long cal(int p,int l,int r){long long ans=0;if (treeNode[p].l==l&&treeNode[p].r==r){ans=ans+treeNode[p].tsum;}else{ans=ans+treeNode[p].csum*(r-l+1);int mid=(treeNode[p].l+treeNode[p].r)/2;if (r<=mid){ans+=cal(2*p,l,r);}else if (l>=mid+1){ans+=cal(2*p+1,l,r);}else{ans+=cal(2*p,l,mid);ans+=cal(2*p+1,mid+1,r);}}return ans;}int main(){int n,q,i,a,b,c;int num;scanf("%d%d",&n,&q);construct(1,1,n);for (i=1;i<=n;i++){scanf("%d",&num);Addnode(1,i,i,num);}char str[2];for (i=1;i<=q;i++){scanf("%s",&str);if (str[0]=='Q'){scanf("%d%d",&a,&b);long long ans=cal(1,a,b);printf("%lld/n",ans);}else if (str[0]=='C'){scanf("%d%d%d",&a,&b,&c);Addnode(1,a,b,c);}}return 0;}