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 100005 struct 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; }