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

来源:互联网 发布:手机防辐射软件 编辑:程序博客网 时间:2024/05/29 05:55

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



阅读全文
1 0
原创粉丝点击