首页 > 代码库 > PKU 2777 Count Color (线段树区间更新)

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;
}