POJ-3468 A Simple Problem with Integers 【线段树:区间更新】
题目链接:http://poj.org/problem?id=3468
题意:对于一串n长的数,进行m次操作,具体如下:
两种操作:’Q’:询问 [ a,b ]区间的和值
‘C’:[ a,b ]区间里面的每个数+x
题解:线段树中的区间更新,如果单点更新 ST_add ( ) 就TLE了,所以要写个延迟标记数组 ad,注意在每次处理的时候标记要清空
延迟标记:每个节点新增加一个标记,记录这个节点是否进行了某种修改(这种修改操作会影响其子节点),对于任意区间的修改,我们先按照区间查询的方式将其划分成线段树中的节点,然后修改这些节点的信息,并给这些节点标记上代表这种修改操作的标记。在修改和查询的时候,如果我们到了一个节点p,并且决定考虑其子节点,那么我们就要看节点p是否被标记,如果有,就要按照标记修改其子节点的信息,并且给子节点都标上相同的标记,同时消掉节点p的标记。
【概念参考了这个大神写的:
http://www.cnblogs.com/TenosDoIt/p/3453089.html#b】
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <map>
#include <cmath>
using namespace std;
typedef long long LL;
const int maxn = 100100;
int n,q,a,b,x;
int num[maxn];
struct ST{
int r,l;
LL ad,sum;
}tr[maxn*4];
char c;
void ST_build(int id,int le,int ri){
tr[id].ad = 0;//----设置标延迟记域
tr[id].l = le,tr[id].r = ri;
if(le==ri){
tr[id].sum = num[le];
return;
}
int mi =(le+ri)/2;
ST_build(id*2,le,mi);
ST_build(id*2+1,mi+1,ri);
tr[id].sum = tr[id*2].sum + tr[id*2+1].sum;
}
void ST_pushDown(int id){ //向子结点传递 id 的add值最后清空
if(tr[id].ad!=0){
tr[id*2].sum += tr[id].ad * (tr[id*2].r - tr[id*2].l + 1);
tr[id*2+1].sum += tr[id].ad * (tr[id*2+1].r - tr[id*2+1].l + 1);
tr[id*2].ad += tr[id].ad;
tr[id*2+1].ad += tr[id].ad;
tr[id].ad = 0; //传递后 当前节点标记清空
}
}
void ST_add(int id,int a,int b,int x){// [a,b] 每个数+x
if( tr[id].l>=a && tr[id].r<=b ) {
tr[id].sum += x*(tr[id].r-tr[id].l + 1);
tr[id].ad += x;
return ;
}
ST_pushDown(id);
int mi = (tr[id].l+tr[id].r)/2;
if(mi>=b) ST_add(id*2,a,b,x); //区间仅在左子树上
else if(mi<a) ST_add(id*2+1,a,b,x); //区间仅在右子树上
else {
ST_add(id*2,a,mi,x); //区间同时在左右子树上
ST_add(id*2+1,mi+1,b,x);
}
tr[id].sum = tr[id*2].sum + tr[id*2+1].sum;
}
LL ST_find(int id,int a,int b){
if(tr[id].l>=a && b >= tr[id].r){
return tr[id].sum;
}
ST_pushDown(id);
int mi = (tr[id].l+tr[id].r)/2;
if( mi>=b ) return ST_find(id*2,a,b);
else if( mi<a ) return ST_find(id*2+1,a,b);
else return ST_find(id*2,a,mi) + ST_find(id*2+1,1+mi,b);
}
int main(){
while(~scanf("%d %d",&n,&q)){
for(int i=1;i<=n;i++){
scanf("%d",&num[i]);
}
ST_build(1,1,n);
while(q--){
getchar();
scanf("%c",&c);
if(c=='Q'){
scanf("%d %d",&a,&b);
printf("%lld\n",ST_find(1,a,b));
}else if(c=='C'){// [a,b] 每个数+x
scanf("%d %d %d",&a,&b,&x);
ST_add(1,a,b,x);
}
}
}
return 0;
}