洛谷P1558 色板游戏

题目背景

阿宝上学了，今天老师拿来了一块很长的涂色板。

题目描述

色板长度为L，L是一个正整数，所以我们可以均匀地将它划分成L块1厘米长的小方格。并从左到右标记为1, 2, ... L。现在色板上只有一个颜色，老师告诉阿宝在色板上只能做两件事：1. "C A B C" 指在A到 B 号方格中涂上颜色 C。2. "P A B" 指老师的提问：A到 B号方格中有几种颜色。学校的颜料盒中一共有 T 种颜料。为简便起见，我们把他们标记为 1, 2, ... T. 开始时色板上原有的颜色就为1号色。 面对如此复杂的问题，阿宝向你求助，你能帮助他吗？

输入输出格式

输入格式：

第一行有3个整数 L (1 <= L <= 100000), T (1 <= T <= 30) 和 O (1 <= O <= 100000). 在这里O表示事件数， 接下来 O 行, 每行以 "C A B C" 或 "P A B" 得形式表示所要做的事情(这里 A, B, C 为整数, 可能A> B)

输出格式：

对于老师的提问，做出相应的回答。每行一个整数。

输入输出样例

输入样例#1： 

2 2 4
C 1 1 2
P 1 2
C 2 2 2
P 1 2

输出样例#1： 

2
1

经典线段树题。。。附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#define LSON rt<<1
#define RSON rt<<1|1
#define DATA(x) a[x].data
#define SIGN(x) a[x].c
#define LSIDE(x) a[x].l
#define RSIDE(x) a[x].r
#define MAXN 100010
using namespace std;
int n,t,m;
struct node{
	long long data,c;
	int l,r;
}a[MAXN<<2];
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
void pushup(int rt){
	DATA(rt)=(DATA(LSON)|DATA(RSON));
}
void pushdown(int rt){
	if(!SIGN(rt)||LSIDE(rt)==RSIDE(rt))
	return;
	SIGN(LSON)=SIGN(rt);
	DATA(LSON)=(1<<SIGN(rt));
	SIGN(RSON)=SIGN(rt);
	DATA(RSON)=(1<<SIGN(rt));
	SIGN(rt)=0;
}
void buildtree(int l,int r,int rt){
	int mid;
	LSIDE(rt)=l;
	RSIDE(rt)=r;
	if(l==r){
		DATA(rt)=(1<<1);
		SIGN(rt)=0;
		return;
	}
	mid=l+r>>1;
	buildtree(l,mid,LSON);
	buildtree(mid+1,r,RSON);
	pushup(rt);
}
void update(int l,int r,long long c,int rt){
	int mid;
	if(l<=LSIDE(rt)&&RSIDE(rt)<=r){
		SIGN(rt)=c;
		DATA(rt)=(1<<c);
		return;
	}
	pushdown(rt);
	mid=LSIDE(rt)+RSIDE(rt)>>1;
	if(l<=mid)update(l,r,c,LSON);
	if(mid<r)update(l,r,c,RSON);
	pushup(rt);
}
long long query(int l,int r,int rt){
	int mid;
	long long ans=0;
	if(l<=LSIDE(rt)&&RSIDE(rt)<=r)
	return DATA(rt);
	pushdown(rt);
	mid=LSIDE(rt)+RSIDE(rt)>>1;
	if(l<=mid)ans=(ans|query(l,r,LSON));
	if(mid<r)ans=(ans|query(l,r,RSON));
	return ans;
}
int main(){
	char ch[2];
	int x,y,k;
	n=read();t=read();m=read();
	buildtree(1,n,1);
	while(m--){
		scanf("%s",ch);x=read();y=read();
		if(x>y)swap(x,y);
		if(ch[0]=='C'){
			k=read();
			update(x,y,k,1);
		}
		if(ch[0]=='P'){
			long long s=query(x,y,1),ans=0;
			while(s){
				if(s&1)
				ans++;
				s>>=1;
			}
			printf("%lld\n",ans);
		}
	}
	return 0;
}