线段树基础操作--单点or区间更新+查询

问题概述:先输入一个数n,表示一个班有n个人,一开始每人手中都有一个糖果,后输入一个数m并进行m次操作,每次操作a、b、c表示从第a个人到第b个人每个人重新分配得到c个糖果,求最后一个班所有人的总糖果数

输入样例:                                     对应输出:

1                                                     24

10

2

1 5 2

5 9 3


区间更新(不能直接更新)要使用懒惰标记的原因:

修改一个区间的值会影响到与这个区间有关的所有子节点和父节点,改变父节点轻而易举,而改变所有子节点则太过于费时,所以要用个数组标记哪些节点的子节点值需要被改变,等到第二次再用到这个节点时再改变其子节点

代码中有注释

#include<stdio.h>
#include<string.h>
int n, tre[444444], temp[444444];
void Readin(int l, int r, int x);
void Update(int l, int r, int x, int a, int b, int val);
void lazy(int l, int r, int x);
int main(void)
{
	int T, m, i, a, b, val;
	scanf("%d", &T);
	while(T--)
	{
		memset(temp, 0, sizeof(temp));
		scanf("%d%d", &n, &m);
		Readin(1, n, 1);
		for(i=1;i<=m;i++)
		{
			scanf("%d%d%d", &a, &b, &val);
			Update(1, n, 1, a, b, val);
		}
		printf("%d\n", tre[1]);
	}
	return 0;
}

void Readin(int l, int r, int x)	/*读入线段树,初始化为每个人一个糖果*/
{
	int m;
	if(l==r)
	{
		tre[x] = 1;
		return;
	}
	m = (l+r)/2;
	Readin(l, m, x*2);
	Readin(m+1, r, x*2+1);
	tre[x] = tre[x*2]+tre[x*2+1];
}

void Update(int l, int r, int x, int a, int b, int val)		/*线段树成段更新*/
{
	int m;
	if(l>=a && r<=b)	/*如果该节点线段被a和b完全包含,则更新这个线段的值*/
	{
		tre[x] = (r-l+1)*val;
		if(l!=r)
			temp[x] = val;		/*temp[x]表示这个节点线段中每个单位值都变成了val*/
		return;					/*temp[x]==0表示这个节点的值没有被改变*/
	}
	if(temp[x]!=0)
		lazy(l, r, x);			/*用于判断这个节点的子节点值是否应该被改变*/
	m = (l+r)/2;
	if(a<=m)
		Update(l, m, x*2, a, b, val);
	if(b>=m+1)
		Update(m+1, r, x*2+1, a, b, val);
	tre[x] = tre[x*2]+tre[x*2+1];
}

void lazy(int l, int r, int x)		/*懒惰标记*/
{
	int m;
	if(l!=r)
		temp[x*2] = temp[x*2+1] = temp[x];	/*继续标记(子节点被改变,那当然子节点的子节点也应该被改变)*/
	m = (l+r)/2;
	tre[x*2] = (m-l+1)*temp[x];
	tre[x*2+1] = (r-m)*temp[x];
	temp[x] = 0;	/*已经改变,temp[x]归0*/
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值