经过昨天六个单点更新线段树水题的洗礼,今天上手快了很多。这道题也是一个线段树的一个入门题
这个题是对一个区间段内的数进行整体的更新,如果对区间的更新还是最终落实到对点的更新,显然复杂度是很高的,因此在处理上需要换一种方式
这次,线段树节点存的内容是其左右子树的颜色信息,如果左右子树颜色相同,那么他的值就是左右子树的颜色值,如果不同,则标记为-1
本题的重心在于update时的处理,如果(L<=l&&R>=r) 即已经找到了要更新的区间对应的节点,则只需修改该节点的值就好,然后返回;如果还没找到,要向下递归的时候,必须先判断这个节点的值,如果不是-1,说明他的两颗子树的颜色都是一样的,那么在向下递归之前,必须先更新两个子节点的值为他自己的值。为什么呢?举个例子.
取n=10,起初所有点都是1.然后更新(1,5)区间的颜色为2,此时,(1,5)这个节点的值变成了2,而(1,5)所有子节点的值都还是1!!,所以当我第二次更新(4,5)区间的颜色为3时,先找到(1,5),然后再向下找到(4,5),将他的值改为3,这时(1,5)的值会因为递归弹出时的push_up被更新为-1.而(1,3)的值还是1!!!,其实(1,3)的值已经在第一步的时候就被更新成2了,只不过我们延迟更新了而已………………所以,在每次向下递归的时候,先得判断自身的值,然后对子节点更新……就像之前这个例子,在(1,5)时,先将两个子节点(1,3)(4,5)都更新为2,然后再向下递归,这样问题就得以解决。
然后还有个地方可以稍作优化,那就是向下递归时,如果区间的值和我要更新的颜色值是相同的,那么就没必要向下递归了
以下是本题的代码:
#include <cstdio>
#include <cstring>
const int MAX = 100010;
int sum[MAX<<2];
void push_up(int rt)
{
sum[rt] = (sum[rt<<1]==sum[rt<<1|1])?sum[rt<<1]:-1;
}
void build(int l,int r,int rt)
{
if(l==r)
{
sum[rt] = 1;
return;
}
int mid = r+l>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
push_up(rt);
}
void update(int p,int L,int R,int l,int r,int rt)
{
if(L<=l&&R>=r)
{
sum[rt] = p;
return;
}
if(sum[rt]>0)
sum[rt<<1] = sum[rt<<1|1] = sum[rt];
int mid = r+l>>1;
if(L<=mid&&sum[rt<<1]!=p) update(p,L,R,l,mid,rt<<1);
if(R>mid&&sum[rt<<1|1]!=p)update(p,L,R,mid+1,r,rt<<1|1);
push_up(rt);
}
int query(int L,int R,int l,int r,int rt)
{
if(sum[rt]>0) return sum[rt]*(r-l+1);
int mid = r+l>>1;
int res = 0;
if(L<=mid) res+=query(L,R,l,mid,rt<<1);
if(R>mid) res+=query(L,R,mid+1,r,rt<<1|1);
return res;
}
int main()
{
int cas;
scanf("%d",&cas);
for(int t=1; t<=cas; t++)
{
int n,q;
scanf("%d%d",&n,&q);
build(1,n,1);
while(q--)
{
int x,y,p;
scanf("%d%d%d",&x,&y,&p);
update(p,x,y,1,n,1);
}
printf("Case %d: The total value of the hook is %d.\n",t,query(1,n,1,n,1));
}
return 0;
}