【线段树（lazy）】poj 2777 Count Color

http://poj.org/problem?id=2777

题意：把长度为L的板子分成L份，标签从1~L，开始时板子的颜色都为1,然后对板子的某一区间涂色，求在某一区间内有多少种不同的颜色？（A可能>B）

分析：区间更新+区间查找

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int NM=100005;
int T[NM*4];
bool f[35];

void Build(int lf,int rg,int rt)
{
	T[rt]=1;
	if(lf==rg) return;
	int mid=(lf+rg)>>1;
	Build(lf,mid,rt<<1);
	Build(mid+1,rg,rt<<1|1);
}

void Update(int x,int y,int vau,int lf,int rg,int rt)
{
	if(T[rt]==vau) return;
	if(x==lf&&y==rg){
		T[rt]=vau;
		return;
	}
	if(T[rt]!=-1){
		T[rt<<1]=T[rt<<1|1]=T[rt];
		T[rt]=-1;
	}
	
	int mid=(lf+rg)>>1;
	if(x>mid) Update(x,y,vau,mid+1,rg,rt<<1|1);
	else if(y<=mid) Update(x,y,vau,lf,mid,rt<<1);
	else {
		Update(x,mid,vau,lf,mid,rt<<1);
		Update(mid+1,y,vau,mid+1,rg,rt<<1|1);
	}
}

void Query(int x,int y,int lf,int rg,int rt)
{
	if(T[rt]!=-1)
		f[T[rt]]=1;
	else {
		int mid=(lf+rg)>>1;
		if(x>mid) Query(x,y,mid+1,rg,rt<<1|1);
		else if(y<=mid) Query(x,y,lf,mid,rt<<1);
		else {
			Query(mid+1,y,mid+1,rg,rt<<1|1);
			Query(x,mid,lf,mid,rt<<1);
		}
	}
}

int main()
{
	int i,n,m,t,x,y,vau,ans;
	char str[3];
	while(scanf("%d%d%d",&n,&t,&m)!=EOF)
	{	
		Build(1,n,1);
		while(m--)
		{
			scanf("%s",str);
			if(str[0]=='C'){
				scanf("%d%d%d",&x,&y,&vau);
				if(x>y) swap(x,y);
				Update(x,y,vau,1,n,1);
			}
			else {
				scanf("%d%d",&x,&y);
				memset(f,0,sizeof(f));
				if(x>y) swap(x,y);
				Query(x,y,1,n,1);
				ans=0;
				for(i=1;i<=t;i++)
					if(f[i]) ans++;
				printf("%d\n",ans);
			}
		}	
	}
	return 0;
}