这是一道模板题。
给定数列 a[1],a[2],…,a[n]你需要依次进行 q 个操作,操作有两类:
1 l r x:给定 l,r,x,对于所有 i∈[l,r],将 a[i] 加上 x(换言之,将 a[l],a[l+1],…,a[r] 分别加上 x);
2 l r:给定 l,r求 ∑ri=la[i]的值(换言之,求 a[l]+a[l+1]+⋯+a[r]的值)。
Input
第一行包含 2 个正整数 n,q表示数列长度和询问个数。保证 1≤n,q≤106。
第二行 n个整数 a[1],a[2],…,a[n]表示初始数列。保证 |a[i]|≤106。
接下来 q 行,每行一个操作,为以下两种之一:
1 l r x:对于所有 i∈[l,r]将 a[i] 加上x;
2 l r:输出 ∑ri=la[i]的值。
保证 1≤l≤r≤n, |x|≤106。
Output
对于每个 2 l r 操作,输出一行,每行有一个整数,表示所求的结果。
Example
样例输入
5 10
2 6 6 1 1
2 1 4
1 2 5 10
2 1 3
2 2 3
1 2 2 8
1 2 3 7
1 4 4 10
2 1 2
1 4 5 6
2 3 4
样例输出
15
34
32
33
50
Hint
对于所有数据,1≤n,q≤106, |a[i]|≤106,1≤l≤r≤n,= |x|≤106。
题解:
对于区间修改,这里用到了懒惰标记,对于一次修改,当找到了目标区间时,就没必要再往下去找他的子树,这时用懒惰标记数组标记此根节点修改的值,若以后用到了他的子树,则改变值带上此懒惰标记的值,这样可以减少时间。
代码:
#include"stdio.h"
#include"string.h"
long long tree[5000010],lazy[5000010];
void build_tree(int node,int start,int end)//建树
{
if(start==end)
scanf("%lld",&tree[node]);
else
{
int mid=(start+end)/2;
build_tree(node*2,start,mid);
build_tree(node*2+1,mid+1,end);
tree[node]=tree[node*2]+tree[node*2+1];
}
}
void pushdown_tree(int node,int l)//懒惰标记
{
if(lazy[node])//如果根节点已经标记,说明其子节点有值没有改变
{
lazy[2*node]+=lazy[node];//左节点的懒惰标记值加上根节点的
lazy[2*node+1]+=lazy[node];//右节点的懒惰标记值加上根节点的
tree[2*node]+=(long long)lazy[node]*(l-(l/2));//计算左右节点的值
tree[2*node+1]+=(long long)lazy[node]*(l/2);
lazy[node]=0;//取消懒惰标记
}
}
void update_tree(int x,int y,int node,int start,int end,int c)
{
if(x<=start&&y>=end)//找到目标区间,不再搜索子节点
{
tree[node]+=(long long)(end-start+1)*c;
lazy[node]+=c;//进行懒惰标记
}
else
{
pushdown_tree(node,end-start+1);//检查是否有懒惰标记
int mid=(start+end)/2;
if(x<=mid)
update_tree(x,y,node*2,start,mid,c);
if(y>mid)
update_tree(x,y,node*2+1,mid+1,end,c);
tree[node]=tree[node*2]+tree[node*2+1];
}
}
void ask_tree(int node,int start,int end,int x)
{
if(start==end)
printf("%lld\n",tree[node]);
else
{
pushdown_tree(node,end-start+1);
int mid=(start+end)/2;
if(mid>=x)
ask_tree(node*2,start,mid,x);
else
ask_tree(node*2+1,mid+1,end,x);
}
}
int main()
{
int m,n;
while(~scanf("%d %d",&n,&m))
{
int i,a,x,y,c;
memset(lazy,0,sizeof(lazy));
memset(tree,0,sizeof(tree));
build_tree(1,1,n);
for(i=0;i<m;i++)
{
scanf("%d",&a);
if(a==1)
{
scanf("%d %d %d",&x,&y,&c);
update_tree(x,y,1,1,n,c);
}
else
{
scanf("%d",&x);
ask_tree(1,1,n,x);
}
}
}
return 0;
}