POJ3468 线段树求和(线段树模板2)
原题链接:POJ3468
解析:本题也是线段树的一种用法,算得上裸题,我用这题来加深对线段树的理解,以及熟悉线段树的模板。郭炜老师的代码是演示指针写法,但我更喜欢用数组来表示,所以改成了我喜欢的模板。
PS:刚刚刚刚刚才知道,这种方法是延迟标记,也就是说当对一端区间的叶子结点进行相同操作的时候,不要急着全部完成,不然递归等操作需要很多,我们可以记录对这个区间将要做什么操作,等查询的时候将所有对此区间的操作一次做完,这样就省的一次一次遍历了。举个例子,就比如对a进行+1操作,进行一千亿亿次,结果和+一千亿亿是一样的,然而效率却天差地别。
失误点:Query函数里的第一行,tree[root].l == s && tree[root].r == e,第一次将tree[root].r 写成tree[root].l,导致答案错误。
代码示例:
#include<iostream>
#include<cstdio>
using namespace std;
long long ans;
struct Node{
int l,r;
long long Inc;
long long sum;
int Mid(){
return (l+r)/2;
}
}tree[400050];
void BuildTree(int root,int l,int r){
tree[root].l = l;
tree[root].r = r;
tree[root].sum = 0;
tree[root].Inc = 0;
if(l == r) return;
int mid = (l+r)/2;
BuildTree(2*root+1,l,mid);
BuildTree(2*root+2,mid+1,r);
}
void Insert(int root,int i,int v){
if(tree[root].l == i && tree[root].r == i){
tree[root].sum = v;
return;
}
tree[root].sum += v;
if(i <= tree[root].Mid()) Insert(2*root+1,i,v);
else Insert(2*root+2,i,v);
}
void Add(int root,int s,int e,long long c){
if(tree[root].l == s && tree[root].r == e){
tree[root].Inc += c;
return;
}
tree[root].sum += c*(e-s+1);
if(e <= tree[root].Mid()) Add(2*root+1,s,e,c);
else if(s > tree[root].Mid()) Add(2*root+2,s,e,c);
else{
Add(2*root+1,s,tree[root].Mid(),c);
Add(2*root+2,tree[root].Mid()+1,e,c);
}
}
long long Query(int root,int s,int e){
if(tree[root].l == s && tree[root].r == e){
return tree[root].sum + tree[root].Inc*(e-s+1);
}
tree[root].sum += tree[root].Inc*(tree[root].r - tree[root].l+1);
Add(2*root+1,tree[root].l,tree[root].Mid(),tree[root].Inc);
Add(2*root+2,tree[root].Mid()+1,tree[root].r,tree[root].Inc);
tree[root].Inc = 0;
if(e <= tree[root].Mid()) return Query(2*root+1,s,e);
else if(s > tree[root].Mid()) return Query(2*root+2,s,e);
else{
return Query(2*root+1,s,tree[root].Mid()) +
Query(2*root+2,tree[root].Mid()+1,e);
}
}
int main()
{
int n,q,h;
scanf("%d%d",&n,&q);
BuildTree(0,1,n);
for(int i = 1;i <= n;i++){
scanf("%d",&h);
Insert(0,i,h);
}
char c[2];
for(int i = 0;i < q;i++){
scanf("%s",c);
if(c[0] == 'Q'){
int s,e;
scanf("%d%d",&s,&e);
printf("%lld\n",Query(0,s,e));
}else{
int s,e,c;
scanf("%d%d%d",&s,&e,&c);
Add(0,s,e,c);
}
}
return 0;
}