Count Color poj2777 线段树

Count Color poj2777 线段树

题意

有一个长木板，现在往上面在一定区间内刷颜色，后来刷的颜色会掩盖掉前面刷的颜色，问每次一定区间内可以看到多少种颜色。

解题思路

这里使用线段树，因为刷颜色可以看作是区间修改，使用lazy标记区间的颜色种类，下传标记后，当前节点的lazy标记就标记为0，然后使用vis数组来标记颜色（颜色种类很少）。剩下的基本就是线段树的模板了。

代码实现

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e5+17;
struct node{
    int l, r, lazy;
}a[maxn<<2];
int vis[35];//颜色种类
int L, T, O, ans;
void build(int rt, int l, int r)
{
    a[rt].l=l;
    a[rt].r=r;
    a[rt].lazy=1;//默认为第一种颜色
    if(l==r){
        return ;
    }
    int mid=(l+r)>>1;
    build(rt<<1, l, mid);
    build(rt<<1|1, mid+1, r);
}
void down(int k)
{
    a[k<<1].lazy=a[k].lazy;
    a[k<<1|1].lazy=a[k].lazy;
    a[k].lazy=0;//表示他的节点下面可能是两种不同的颜色
}
void update(int rt, int L, int R, int x)
{
    if(L<=a[rt].l && a[rt].r<=R)
    {
        a[rt].lazy=x;
        return ; 
    }
    int mid=(a[rt].l+a[rt].r)>>1;
    if(a[rt].lazy !=0 ) //记得一定要下传标记 
        down(rt); 
    if(L<=mid) update(rt<<1, L, R, x);
    if(R>mid) update(rt<<1|1, L, R, x);
}
void query(int rt, int L, int R)
{
    if(a[rt].lazy!=0) //如果不为零就可以进行判断，因为下面的也是这种颜色
    {
        if(!vis[a[rt].lazy])// 看是否之前标记过
        {
            ans++; //没有标记就加一
            vis[a[rt].lazy]=1; //标记
        }
        return ;
    }
    int mid=(a[rt].l+a[rt].r)>>1;
    if(a[rt].lazy!=0)
        down(rt);
    if(L<=mid) query(rt<<1, L, R);
    if(R>mid) query(rt<<1|1, L, R);
}
int main()
{
    int l, r, c;
    char s[4];
    while(scanf("%d%d%d", &L, &T, &O)!=EOF)
    {
        build(1, 1, L);
        for(int i=1; i<=O; i++)
        {
            scanf("%s", s);
            if(s[0]=='C')
            {
                scanf("%d%d%d", &l, &r, &c);
                if(l > r){
                    swap(l, r);
                }
                update(1, l, r, c);
            }
            else {
                scanf("%d%d", &l, &r);
                if(l > r) {
                    swap(l, r); 
                }
                memset(vis, 0, sizeof(vis));
                ans=0;
                query(1, l, r);
                printf("%d\n", ans);
            }
        }   
    }
    return 0;
}

转载于:https://www.cnblogs.com/alking1001/p/11316507.html