关闭

hdu5023 A Corrupt Mayor's Performance Art(线段树+区间更新)

104人阅读 评论(0) 收藏 举报

hdu5023

题目

P :A到B区间图上第C种颜色;
Q:A到B所有点存在的颜色种类;

思路

因为只有30种颜色,可以用位运算进行状态压缩,线段树就是区间更新,区间查询。

代码

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

#define lson l, m , rt << 1
#define rson m + 1 ,r , rt << 1 | 1

using namespace std;

typedef long long ll;

const int maxn=1000010;

int n,m;
ll col[maxn<<2];
ll sum[maxn<<2];

void PushDown(int rt)
{
    if(col[rt])
    {
        col[rt<<1]=col[rt];
        col[rt<<1|1]=col[rt];
        sum[rt<<1]=col[rt];
        sum[rt<<1|1]=col[rt];
        col[rt]=0;
    }
}

void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]|sum[rt<<1|1];
}

void build(int l,int r,int rt)
{
    col[rt]=0;
    if(l==r)
    {
        sum[rt]=2;
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt);
}

void update(int c,int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        col[rt]=1<<(c-1);
        sum[rt]=1<<(c-1);
        return;
    }
    PushDown(rt);
    int m=(l+r)>>1;
    if(L<=m) update(c,L,R,lson);
    if(R>m) update(c,L,R,rson);
    PushUp(rt);
}

ll query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return sum[rt];
    }
    ll ans=0;
    PushDown(rt);
    int m=(l+r)>>1;
    if(L<=m) ans|=query(L,R,lson);
    if(R>m) ans|=query(L,R,rson);
    return ans;
}

int main()
{
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        if(n==0&&m==0)
            break;
        build(1,n,1);
        char op;
        int a,b,c;
        for(int i=0; i<m; i++)
        {
            getchar();
            scanf("%c",&op);
            if(op=='P')
            {
                scanf("%d %d %d",&a,&b,&c);
                update(c,a,b,1,n,1);
            }
            else if(op=='Q')
            {
                scanf("%d %d",&a,&b);
                ll sum=query(a,b,1,n,1);
                int cnt=0;
                for(int j=1; j<=30; j++)
                {
                    if(sum&(1<<(j-1)))
                    {
                        if(cnt!=0)
                            printf(" %d",j);
                        else
                            printf("%d",j);
                        cnt++;
                    }
                }
                printf("\n");
            }
        }

    }
    return 0;
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:21173次
    • 积分:1679
    • 等级:
    • 排名:千里之外
    • 原创:155篇
    • 转载:0篇
    • 译文:0篇
    • 评论:5条