hdu Just a Hook 线段树成段更新.cpp

#include <stdio.h>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define maxn 111111
int col[maxn<<2];
int sum[maxn<<2];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt,int m)//m为该区间的长度
{
    if(col[rt])//表示该节点所代笔的区间被完全染色.将父亲节点的sum和col传递到它的子节点
    {
        col[rt<<1]=col[rt<<1|1]=col[rt];
        sum[rt<<1]=(m-(m>>1))*col[rt];//切记不可写成(m>>1),因为比如当m=5时,m>>1为2,而前者为3.
        sum[rt<<1|1]=(m>>1)*col[rt];
        col[rt]=0;//该操作时递归操作,父亲节点向子节点传递的时候,本身的标记被抹去,以免重复搜索
    }
}
void build(int l,int r,int rt)
{
    col[rt]=0;
    sum[rt]=1;
    if(l==r) return ;
    int mid=(r+l)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int a,int b,int c,int l,int r,int rt)//注意和单点求和的区别.单点求和先修改叶子节点的值,然后递归修改祖先.
{//成段修改可将查询和求和的过程合并
    if(a<=l&&b>=r)//表示该节点的区间被完全覆盖
    {
        col[rt]=c;
        sum[rt]=(r-l+1)*col[rt];
        return ;
    }
    pushdown(rt,r-l+1);//pushdown(rt,b-a+1);是错的.(a,b)是要查找的区间,不是该节点上的区间.
    int mid=(r+l)>>1;
    if(a<=mid) update(a,b,c,lson);
    if(b>mid) update(a,b,c,rson);//切记,右区间是mid+1,所以b>mid..b>=mid  运行错误.
    pushup(rt);
}
int main()
{
     int t,n,m,k,a,b;
     int cases=1;
     scanf("%d",&t);
     while(t--)
     {
         scanf("%d%d",&n,&m);
         build(1,n,1);//我傻逼了!!
         while(m--)
         {
             scanf("%d%d%d",&a,&b,&k);//sum[1]是最远的祖先,也是整个数组的总和
             update(a,b,k,1,n,1);
         }
         printf("Case %d: The total value of the hook is %d.\n",cases++,sum[1]);
     }
     return 0;
}