HDU-1698 Just a Hook

简单的线段树实现

每个节点维护区间和，最后输出根节点sum就可以了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

using namespace std;
typedef long long ll;
const int maxn=100000+10; //数组大小

int lazy[maxn<<2]; //懒惰标记
int sum[maxn<<2];
int value[maxn]; //初始值

void PushUp(int rt) //向上更新父节点
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void PushDown(int rt,int m) //向下更新
{
    if(lazy[rt]) //懒惰标记，这里的语句根据更新函数而变
    {
        lazy[rt<<1]=lazy[rt];
        lazy[rt<<1|1]=lazy[rt];
        sum[rt<<1]=lazy[rt]*(m-(m>>1));
        sum[rt<<1|1]=lazy[rt]*(m>>1);
        lazy[rt]=0;
    }
}

void build(int l,int r,int rt) //建树
{
    lazy[rt]=0;
    if(l==r)
    {
        sum[rt]=value[l];
        return;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    PushUp(rt);
}

void update(int L,int R,int c,int l,int r,int rt) //将L~R区间变为c，可以根据需要修改更新函数
{
    if(L<=l&&R>=r)
    {
        lazy[rt]=c;
        sum[rt]=c*(r-l+1);
        return;
    }
    PushDown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m) update(L,R,c,l,m,rt<<1);
    if(m<R) update(L,R,c,m+1,r,rt<<1|1);
    PushUp(rt);
}

int query(int L,int R,int l,int r,int rt) //查询L~R
{
    if(L<=l&&R>=r)
    return sum[rt];
    PushDown(rt,r-l+1);
    int m=(l+r)>>1;
    int ret=0;
    if(L<=m) ret+=query(L,R,l,m,rt<<1);
    if(m<R) ret+=query(L,R,m+1,r,rt<<1|1);
    return ret;
}

int n,m;
int a,b,c;

int main()
{
    //freopen("/home/zlwang/test.txt","r",stdin);
    int T,kase=0;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        value[i]=1;
        build(1,n,1);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            update(a,b,c,1,n,1);
        }
        printf("Case %d: The total value of the hook is %d.\n",++kase,query(1,n,1,n,1));
    }
    return 0;
}