poj2777 线段树

最新推荐文章于 2022-02-05 14:40:30 发布

Sun_shiney

最新推荐文章于 2022-02-05 14:40:30 发布

阅读量264

点赞数

分类专栏：线段树

本文链接：https://blog.youkuaiyun.com/Sun_shiney/article/details/52194526

版权

线段树专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种基于线段树的区间染色查询算法，通过将颜色信息压缩为二进制来快速处理区间内的染色操作及不同颜色数量的查询。适用于颜色种类较少的情况。

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

题意：给一些线段区间染色，然后问你一段区间的不同颜色的个数

分析：因为颜色比较小可以压缩为二进制表示

细节详见代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int maxn = 1e5 + 20;
struct InterxTree
{
	int sum[maxn<<2],set[maxn<<2];
void build()
{
    sum[1]=1,set[1]=1;
}
void pushDown(int rt)
{
    sum[rt<<1]=sum[rt<<1|1]=1<<(set[rt]-1);
    set[rt<<1]=set[rt<<1|1]=set[rt];
    set[rt]=-1;
}
void maintain(int rt)
{
	sum[rt]=sum[rt<<1]|sum[rt<<1|1];
}
void update(int L,int R,int v,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        set[rt]=v;
        sum[rt]=1<<(v-1);
        return ;
    }
    if(~set[rt])
        pushDown(rt);
    int mid=(l+r)>>1;
    if(L<=mid) update(L,R,v,lson);
    if(R>mid) update(L,R,v,rson);
    maintain(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return sum[rt];
    }
    if(~set[rt])	         // 查询时候不用pushdown 减少时间浪费 
    {
    	return 1<<(set[rt] -1);
	}
    int mid=(l+r)>>1;
    int ans = 0;
    if(L<=mid) ans |=query(L,R,lson);
    if(R>mid) ans |=query(L,R,rson);
    return ans;
}
}tree;
int main()
{
//	freopen("1.txt","r",stdin);
    int l,t,o;
    while(~scanf("%d %d %d",&l,&t,&o))
    {
    	tree.build();
    while(o--)
    {
        char d = getchar();
        int a,b,c;
        while(!isalpha(d)) d = getchar();
        if(d=='C')
        {
            scanf("%d %d %d",&a,&b,&c);
            if(b<a)
                swap(a,b);
            tree.update(a,b,c,1,l,1);
        }
        else
        {
            scanf("%d %d",&a,&b);
            if(b<a)
                swap(a,b);
            printf("%d\n",__builtin_popcount(tree.query(a,b,1,l,1)));    //  数1的个数 
        }
    }
}
    return 0;
}