POJ 2777 Count Color

题目链接：http://poj.org/problem?id=2777

题意：给出一面长度为len的墙，一开始的颜色是1，进行2种操作，一是将[l,r]涂成颜色p，二是询问[l,r]区间一共有多少种颜色

思路：对区间进行修改和询问，很容易想到用线段树，但是如何保存有多少种颜色呢，因为颜色少于30种（我又看了题解），所以可以用位运算进行状态压缩，剩下的就是线段树普通的区间修改和更新了（写题的时候花式出错，晚上太亢奋了吧……）

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 100030
using namespace std;

struct Tree
{
int l,r,date;
}tree[maxn*3];

int lazy[maxn*3];

void build(int root,int l,int r)
{
tree[root].l=l;
tree[root].r=r;
if (l==r)
{
tree[root].date=(1<<1);
return;
}
int mid=(l+r)>>1;
build (root<<1,l,mid);
build (root<<1|1,mid+1,r);
tree[root].date=tree[root<<1].date|tree[root<<1|1].date;
}

void update(int root,int l,int r,int val)
{

if (tree[root].l>=l && tree[root].r<=r)
{
tree[root].date=(1<<val);
lazy[root]=(1<<val);
return;
}

if (lazy[root]!=0)
{
tree[root<<1].date=tree[root].date;
tree[root<<1|1].date=tree[root].date;
lazy[root<<1]=lazy[root];
lazy[root<<1|1]=lazy[root];
lazy[root]=0;
}
int mid=(tree[root].l+tree[root].r)>>1;

if (l<=mid) update(root<<1,l,r,val);
if (r>mid) update(root<<1|1,l,r,val);
tree[root].date=tree[root<<1].date|tree[root<<1|1].date;
}

int que(int root,int l,int r)
{

if (lazy[root]!=0) return lazy[root];

if (tree[root].l==l && tree[root].r==r)
return tree[root].date;

int mid=(tree[root].l+tree[root].r)>>1;
if (r<=mid) return que(root<<1,l,r);
else if (l>mid) return que(root<<1|1,l,r);
else return que(root<<1,l,mid)|que(root<<1|1,mid+1,r);

}

int main()
{
int len,n,m;
while (scanf("%d%d%d",&len,&n,&m)!=EOF)
{
memset(lazy,0,sizeof(lazy));
build (1,1,len);
for (int i=0;i<m;i++)
{
char tem;
cin>>tem;
if (tem=='C')
{
int a,b,p;
scanf("%d%d%d",&a,&b,&p);
if (a>b) swap(a,b);
update(1,a,b,p);
}
else
{
int a,b,res=0,cl;
scanf("%d%d",&a,&b);
if (a>b) swap(a,b);
cl=que(1,a,b);
//  cout<<":"<<cl<<endl;
for (int i=1;i<=n;i++)
{
if ((cl&(1<<i))!=0) res++;
}
printf("%d\n",res);
}
}
}
}