hdu 5023 线段树+状压-优快云博客

本文介绍了一种利用线段树和状压技术解决片段着色问题的方法，包括两种操作：改变指定片段的颜色和查询指定片段内的所有颜色。详细解释了如何构建线段树、插入颜色、查询颜色，并提供了代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://acm.hdu.edu.cn/showproblem.php?pid=5023

在片段上着色，有两种操作，如下：
第一种：P a b c 把 a 片段至 b 片段的颜色都变为 c 。
第二种：Q a b 询问 a 片段至 b 片段有哪些颜色，把这些颜色按从小到大的编号输出，不要有重复
片段上默认的初始颜色为编号2的颜色。

颜色30种，状压；线段树进行更新和询问

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 1000005;
int ans[35];
void cntbit(int x)
{
    int cnt = 0;
    for(int i = 1;i <= 30;++i){
        if(x&(1<<(i-1)))
            ans[cnt++] = i;
    }
    for(int i = 0;i < cnt;++i)
        printf("%d%c",ans[i]," \n"[i == cnt-1]);
    return;
}
struct node
{
    int l,r;
    int color;//color用位1个数记录那些颜色被涂抹
    bool leaf;//当线段恰好覆盖一个节点的区间时就直接对该节操作而不再向下操作
    //对于这种线段树，要在获得整块区间时停止并把该节点的leaf改为true
}s[N*3];

void build(int l,int r,int root)
{
    s[root].l = l;
    s[root].r = r;
    s[root].leaf = false;
    if(l == r){
        return;
    }
    int mid = (l+r)>>1;
    build(l,mid,root<<1);
    build(mid+1,r,(root<<1)+1);
    return;
}
void insert(int l,int r,int root,int color)
{
    if(l == s[root].l && s[root].r == r){
        s[root].color = color;
        s[root].leaf = true;
        return;
    }
    if(s[root].leaf){
        s[root].leaf = false;
        s[root<<1].leaf = true;
        s[(root<<1)+1].leaf = true;
        s[root<<1].color = s[root].color;
        s[(root<<1)+1].color = s[root].color;
    }
    int mid = (s[root].l+s[root].r)>>1;
    if(mid >= r)
        insert(l,r,root<<1,color);
    else if(l > mid)
        insert(l,r,(root<<1)+1,color);
    else{
        insert(l,mid,root<<1,color);
        insert(mid+1,r,(root<<1)+1,color);
    }
    s[root].color = s[root<<1].color | s[(root<<1)+1].color;
}
int query(int l,int r,int root)
{
    //s[root].leaf == true的判断不能丢
    if( s[root].leaf || (s[root].l == l && s[root].r == r)){
        return s[root].color;
    }
    int mid = (s[root].l + s[root].r)>>1;
    if(r <= mid){
        return query(l,r,root<<1);
    }else if(l > mid){
        return query(l,r,(root<<1)+1);
    }else{
        return query(l,mid,root<<1) | query(mid+1,r,(root<<1)+1);
    }
}
int main()
{
    int n,q;
    while(~scanf("%d%d",&n,&q),n|q){
        build(1,n,1);
        insert(1,n,1,2);
        int l,r,col;
        char ss[5];
        while(q--){
            scanf("%s%d%d",ss,&l,&r);
            if(l > r)
                swap(l,r);
            if(ss[0] == 'P'){
                scanf("%d",&col);
                insert(l,r,1,1<<(col-1));
            }
            else{
                cntbit(query(l,r,1));
            }
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/zibaohun/p/4046812.html