POJ3468 A Simple Problem with Integers 线段树求解区间和

Problem Address：http://poj.org/problem?id=3468

 

【前言】

 

其实修改区域查询区域的情况是可以用树状数组求解的。

 

参考地址：http://blog.youkuaiyun.com/Human_CK/archive/2011/06/13/6542370.aspx

 

出于学习线段树的目的，所以用了线段树。

 

【思路】

 

线段树前面也有写过，这里不多说了。

 

主要点是，每个节点记录对应区间的总和并维护其准确性，并且用变量add记录对这个节点区间每个节点的相同增值。

 

更新的时候，从上往下更新。

 

如果吻合区间则增值增加。否则递归求解。

 

查询的时候，从上往下查。

 

如果刚好吻合，则返回其上对这个区间所有增值的总和+这个区间的总和。

 

如果不吻合则递归下去。

 

【代码】

 

#include <iostream> using namespace std; const int maxn = 100000; __int64 num[maxn+10]; struct node { int l, r; node *pl, *pr; __int64 sum; __int64 add; }tree[maxn*3]; int total; node *new_node() { node *p = &tree[total]; total++; memset(p, 0, sizeof(node)); return p; } node *build(int l, int r) { node *root = new_node(); root->l = l; root->r = r; if (r>l) { int mid = (l+r)/2; root->pl = build(l, mid); root->pr = build(mid+1, r); } if (l==r) root->sum = num[l]; else root->sum = root->pl->sum + root->pr->sum; return root; } void update(node *root, int l, int r, int v) { if (root->l==l && root->r==r) { root->add += v; return; } root->sum += (r-l+1)*v; int mid = (root->l+root->r)/2; if (r<=mid) update(root->pl, l, r, v); else if (l>mid) update(root->pr, l, r, v); else { update(root->pl, l, mid, v); update(root->pr, mid+1, r, v); } } __int64 getsum(node *root, int l, int r, __int64 temp) { temp += root->add; if (root->l==l && root->r==r) return root->sum + (r-l+1)*temp; int mid = (root->l+root->r)/2; if (r<=mid) return getsum(root->pl, l, r, temp); else if (l>mid) return getsum(root->pr, l, r, temp); else return getsum(root->pl, l, mid, temp) + getsum(root->pr, mid+1, r, temp); } int main() { int n,m; int a,b,c; char cmd[2]; int i; node *root; scanf("%d %d", &n, &m); for (i=1; i<=n; i++) scanf("%I64d", &num[i]); total = 0; root = build(1, n); for (i=0; i<m; i++) { scanf("%s", cmd); if (cmd[0]=='Q') { scanf("%d %d", &a, &b); printf("%I64d/n", getsum(root, a, b, 0)); } else { scanf("%d %d %d", &a, &b, &c); update(root, a, b, c); } } return 0; }

 

【P.S】

 

话说线段树的代码也不是特别复杂= =