HDU 1698 Just a Hook 线段树区间更新

经过昨天六个单点更新线段树水题的洗礼，今天上手快了很多。这道题也是一个线段树的一个入门题

这个题是对一个区间段内的数进行整体的更新，如果对区间的更新还是最终落实到对点的更新，显然复杂度是很高的，因此在处理上需要换一种方式

这次，线段树节点存的内容是其左右子树的颜色信息，如果左右子树颜色相同，那么他的值就是左右子树的颜色值，如果不同，则标记为-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;
}