码迷,mamicode.com
首页 > 其他好文 > 详细

Count Color(poj 2777)

时间:2016-06-28 21:54:20      阅读:182      评论:0      收藏:0      [点我收藏+]

标签:

题意:

给一个固定长度为L的画板

有两个操作:

C A B C:区间AB内涂上颜色C。

P A B:查询区间AB内颜色种类数。

分析:显然是要用线段树来操作的,设定一个sum[]来维护一个区间内的颜色,若为-1,则说明不是叶子节点,还应在往底下找;

       同时在查询时维护一个vis[]来记录在这个区间内哪个颜色被使用过了。

技术分享
#include<cstdio>
#include<iostream>
#include<cstring>
#define lson l,m,now*2
#define rson m+1,r,now*2+1
#define M 400010
#define N 35
using namespace std;
int vis[N],sum[M];//当sum[i]=-1是,说明不是底下至少有2种颜色,否则sum[i]是i的颜色 
void build(int l,int r,int now)
{
    sum[now]=1;
    if(l==r)return;
    int m=(l+r)/2;
    build(lson);
    build(rson);
}
void push_down(int now)
{
    sum[now*2]=sum[now];
    sum[now*2+1]=sum[now];
    sum[now]=-1;
}
void change(int x,int y,int v,int l,int r,int now)
{
    if(l>=x&&r<=y)
    {
        sum[now]=v;
        return;
    }
    if(sum[now]==v)return;
    if(sum[now]!=-1)push_down(now);
    int m=(l+r)/2;
    if(x<=m)change(x,y,v,lson);
    if(y>m)change(x,y,v,rson);
}
void query(int x,int y,int l,int r,int now)
{
    if(sum[now]!=-1)
    {
        vis[sum[now]]=1;
        return;
    }
    int m=(l+r)/2;
    if(x<=m)query(x,y,lson);
    if(y>m)query(x,y,rson);
}
int main()
{
    int n,m,p;
    scanf("%d%d%d",&n,&m,&p);
    build(1,n,1);
    for(int i=1;i<=p;i++)
    {
        char c;
        cin>>c;
        if(c==C)
        {
            int a,b,v;
            scanf("%d%d%d",&a,&b,&v);
            if(a>b)swap(a,b);
            change(a,b,v,1,n,1);
        }
        else
        {
            memset(vis,0,sizeof(vis));
            int a,b,tot=0;
            scanf("%d%d",&a,&b);
            if(a>b)swap(a,b);
            query(a,b,1,n,1);
            for(int i=1;i<=30;i++)
              if(vis[i])tot++;
            printf("%d\n",tot);
        }
    }
    return 0;
}
View Code

 

Count Color(poj 2777)

标签:

原文地址:http://www.cnblogs.com/harden/p/5624932.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!