HDU - 1698 Just a Hook(线段树)

成段修改。

#include <iostream>
#include <cstdio>
using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int maxn=111111;
int sum[maxn<<2], col[maxn<<2];

void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l, int r, int rt)
{
    col[rt]=0;sum[rt]=1;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void pushdown(int rt, int m)
{
    if(!col[rt])return ;
    col[rt<<1]=col[rt<<1|1]=col[rt];
    sum[rt<<1]=(m-(m>>1))*col[rt];
    sum[rt<<1|1]=(m>>1)*col[rt];
    col[rt]=0;
}
void update(int L, int R, int v, int l, int r, int rt)
{
    if(L<=l && R>=r)
    {
        sum[rt]=v*(r-l+1);
        col[rt]=v;
        return ;
    }
    int mid=(l+r)>>1;
    pushdown(rt, r-l+1);
    if(L<=mid)update(L, R, v, lson);
    if(R>mid)update(L, R, v, rson);
    pushup(rt);
}
int main()
{
    int T, kase=0, n, q, a, b, v;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &n, &q);
        build(1, n, 1);
        while(q--)
        {
            scanf("%d%d%d", &a, &b, &v);
            update(a, b, v, 1, n, 1);
        }
        printf("Case %d: The total value of the hook is %d.\n", ++kase, sum[1]);
    }
    return 0;
}