问题概述:先输入一个数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*/
}