PKU 2777 Count Color （线段树区间更新）

题意： 给你三个数：L (1 <= L <= 100000), T (1 <= T <= 30) and O (1 <= O <= 100000)，表示有一长度为L的板（1~L），

有T种颜色（1~T），然后有O个操作，初始板1~L的颜色为1，"C A B C"表示在区间A，B图上C颜色， "P A B" 表示询问

A，B区间有几种不同的颜色。

#include <stdio.h>  
#include <iostream>  
#include <algorithm>  
#include <string.h>  
#include <math.h>  
#define M 100000  
#define LL long long  
using namespace std;  

struct Tree
{
	int l,r,c;//c代表颜色，若区间（l，r）的颜色不止一种则c=-1；
}tree[M*3];
int color[35];//在查询时，若[A,B]区间内出现过某种颜色k,则color[k]=true;
void build(int id,int l,int r)
{
	tree[id].l=l;
	tree[id].r=r;
	tree[id].c=1;//初始颜色为1
	if(l==r) return ;
	build(id*2,l,(l+r)/2);
	build(id*2+1,(l+r)/2+1,r);
}
void update(int id,int l,int r,int co)
{
	if(tree[id].l>=l&&tree[id].r<=r)//（l，r）完全覆盖id节点
	{
		tree[id].c=co;
		return;
	}
	if(tree[id].c>0)//一直wa在这里
	{
		tree[id*2].c=tree[id*2+1].c=tree[id].c;
		tree[id].c=-1;
	}
	tree[id].c=-1;//没有完全覆盖
	int mid=(tree[id].l+tree[id].r)/2;
	if(mid>=r)
		update(2*id,l,r,co);
	else if(mid<l)
		update(2*id+1,l,r,co);
	else 
	{
		update(2*id,l,mid,co);
		update(2*id+1,mid+1,r,co);
	}
}
void P(int id,int l,int r)
{
	if(tree[id].c>0)
	{
		color[tree[id].c]=true;
		return;
	}
	if(tree[id].l==tree[id].r) return;
	int mid=(tree[id].l+tree[id].r)/2;
	if(mid>=r)
		P(id*2,l,r);
	else if(mid<l)
		P(id*2+1,l,r);
	else 
	{
		P(id*2,l,mid);
		P(id*2+1,mid+1,r);
	}
}
int Ans(int t)
{
	int ans=0;
	for(int i=1;i<=t;i++)
		if(color[i])
			ans++;
	return ans;
}
int n,t,m;
int main()
{
	scanf("%d%d%d",&n,&t,&m);
	char s[2];
	int a,c,b;
	build(1,1,n);
	while(m--)
	{
		scanf("%s",s);
		if(s[0]=='C')
		{
			scanf("%d%d%d",&a,&b,&c);
			if(a>b) swap(a,b);
			update(1,a,b,c);
		}
		else 
		{
			scanf("%d%d",&a,&b);
			if(a>b) swap(a,b);
			memset(color,false,sizeof(color));
			P(1,a,b);
			printf("%d\n",Ans(t));
		}
	}

	return 0;
}