POJ2777 Count Color【线段树】

很显然，这是一道线段树的题，但难点就是怎么保存区间的颜色数，想了很久，曾经考虑过二进制，可是我不太懂二进制，所以就不想了，其实不用二进制也可以，可是脑子笨，不会用数组来保存颜色数，到现在还是不太会，，，，

进入正题吧，这道题颜色最多只有30种，所以可以用二进制的方法来保存每一种不同的颜色，例如 2^0，2^1，2^2，2^4……2^29来表示颜色1,2,3……30.为什么减一呢？因为二进制中2^0表示1，第一位数。怎么表示那颜色数呢？两者位或，然后记录二进制中1的数就是颜色数了，（二进制真是一个强大的工具啊，可是不太懂）。例如2表示10,3表示100，则2|3表示110，有两个1，表示两种颜色。运用位或可以去重（何为位或？例如110|010=110,1010|0110=1110）

用二进制保存颜色数后，剩下的就是线段树的工作了，注意因为是区间更新，所以要用到lazy思想，就是边走边向下传递。。。

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

using namespace std;
#define N 100005
int s[N<<2];
int a[N<<2];

void up(int rt)
{
   s[rt]=s[rt<<1]|s[rt<<1|1];   //位或
}
void down(int rt)
{
    if(a[rt])
    {
        a[rt<<1]=a[rt<<1|1]=a[rt];
        s[rt<<1]=s[rt<<1|1]=1<<(a[rt]-1);
        a[rt]=0;
    }
}
void build(int l,int r,int rt)
{
    a[rt]=0;
    if(l==r)
    s[rt]=1;
    else
    {
        int m=(l+r)>>1;
        build(l,m,rt<<1);
        build(m+1,r,rt<<1|1);
        up(rt);
    }
}
void updata(int left,int right,int l,int r,int rt,int p)
{
    if(left<=l&&right>=r)
    {
        a[rt]=p;
        s[rt]=1<<(p-1);
        return ;
    }
    down(rt);
    int m=(l+r)>>1;
    if(left<=m)
    updata(left,right,l,m,rt<<1,p);
    if(right>m)
    updata(left,right,m+1,r,rt<<1|1,p);
    up(rt);
}
int query(int left,int right,int l,int r,int rt)
{
    if(left<=l&&right>=r)
    {
        return s[rt];
    }
    down(rt);
    int ans=0;
    int m=(l+r)>>1;
    if(left<=m)
    ans|=query(left,right,l,m,rt<<1);
    if(right>m)
    ans|=query(left,right,m+1,r,rt<<1|1);
    return ans;
}
int main()
{
    int l,t,o,a,b,c;
    char op[10];
    scanf("%d%d%d",&l,&t,&o);
    build(1,l,1);
    while(o--)
    {
        scanf("%s",op);
        if(op[0]=='C')
        {
            scanf("%d%d%d",&a,&b,&c);
            if(a>b)
            {
                int temp=a;a=b;b=temp;
            }
            updata(a,b,1,l,1,c);
        }
        else
        {
            scanf("%d%d",&a,&b);
            if(a>b)
            {
                int temp=a;a=b;b=temp;
            }
            //cout<<a<<" "<<b<<endl;
            int ans=query(a,b,1,l,1);
            //cout<<ans<<endl;
            int color=0;
            for(int i=0;i<t;i++)    //ans中1的个数
            if(ans&(1<<i))
            color++;
            printf("%d\n",color);
        }
    }
    return 0;
}

相关链接：
http://poj.org/problem?id=2777