线段树(区间修改,单点查询)懒惰标记

这是一道模板题。

给定数列 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;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值