【题解】hdu1698 线段树区间修改

题目链接
线段树维护一段区间内钩子的颜色(材料)是否相同

#include<cstdio>
#define lc (o<<1)
#define rc (o<<1|1)
#define mid (l+r>>1)
const int N=1e5+5;
int s[N*4];//s为-1是杂色,s不为-1是纯色,可以直接出答案 
void build(int o,int l,int r)
{
    if(l==r)return;
    s[o]=1;
    build(lc,l,mid);build(rc,mid+1,r);
}
void update(int o,int l,int r,int nl,int nr,int v)
{
    if(s[o]==v)return;//相同不用改 
    if(l==nl&&r==nr)
    {
        s[o]=v;
        return;
    }
    if(s[o]!=-1){//纯色 
        s[lc]=s[rc]=s[o];
        s[o]=-1;//变为杂色 
    }
    if(nr<=mid)update(lc,l,mid,nl,nr,v);
    else if(nl>mid)update(rc,mid+1,r,nl,nr,v);
    else{
        update(lc,l,mid,nl,mid,v);update(rc,mid+1,r,mid+1,nr,v);
    }
}
int query(int o,int l,int r)
{
    if(s[o]!=-1)return (r-l+1)*s[o];//纯色 
    else return query(lc,l,mid)+query(rc,mid+1,r);
}
int main()
{
    //freopen("in.txt","r",stdin);
    int t,k,n,i,q,x,y,z;
    scanf("%d",&t);
    for(k=1;k<=t;k++){
        scanf("%d%d",&n,&q);
        build(1,1,n);
        for(i=1;i<=q;i++){
            scanf("%d%d%d",&x,&y,&z);
            update(1,1,n,x,y,z);
        }
        printf("Case %d: The total value of the hook is %d.\n",k,query(1,1,n));
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41958841/article/details/81608099
今日推荐