线段树维护区间颜色数量 色板游戏(洛谷P1558)

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,这样的话需要你交换A和B)

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

输入 #1

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

输出 #1

2
1

显然的线段树,但这是区间染色的线段树,要结合位运算符<或 | >来解决,真的佩服想到这种方法的人;维护每个区间的或位运算值,算出来每个二进制位如果为1那么就代表一个颜色;具体可以模拟一下,1001|0001=1001,代表这两个区间合并只有两种颜色;1001|0110=1111,代表这两个区间合并有4种颜色;要注意每次修改把颜色传化为二进制传进去,每个颜色都有自己二进制表示;最后统计sum有几个1就完事了;
代码:

#include
using namespace std;
int x,y,z;
int sum;
struct Node{
     
	int l,r,f,color;
}tree[400100];
inline void build(int k,int ll,int rr){
     
	tree[k].l=ll,tree[k].r=rr,tree[k].f=0;
	if(ll==rr){
     
		tree[k].color=1;
		return;
	}
	int m=(ll+rr)>>1;
	build(k<<1,ll,m);
	build(k<<1|1,m+1,rr);
	tree[k].color=tree[k<<1].color|tree[k<<1|1].color;
}
inline void pd(int k){
     
	if(tree[k].f){
     
		tree[k<<1].f=tree[k].f;
		tree[k<<1|1].f=tree[k].f;
		tree[k<<1].color=tree[k].color;
		tree[k<<1|1].color=tree[k].color;
		tree[k].f=0;
		return;
	}
}
inline void change(int k){
     
	if(tree[k].l>=x&&tree[k].r<=y){
     
		tree[k].color=z;
		tree[k].f=1;
		return;
	}
	pd(k);
	int m=(tree[k].l+tree[k].r)>>1;
	if(x<=m) change(k<<1);
	if(y>m) change(k<<1|1);
	tree[k].color=tree[k<<1].color|tree[k<<1|1].color;
}
inline void ask(int k){
     
	if(tree[k].l>=x&&tree[k].r<=y){
     
		sum|=tree[k].color;
		return;
	}
	pd(k);
	int m=(tree[k].l+tree[k].r)>>1;
	if(x<=m) ask(k<<1);
	if(y>m) ask(k<<1|1);
}
inline int solve(int k){
     
	int ans=0;
	while(k){
     
		if(k&1){
     
			ans++;
		}
		k>>=1;
	}
	return ans;
}
int main(){
     
	int L,T,O;
	scanf("%d%d%d",&L,&T,&O);
	build(1,1,L);
	char s[5];
	for(int i=1;i<=O;i++){
     
		scanf("%s",s);
		if(s[0]=='C'){
     
			scanf("%d%d%d",&x,&y,&z);
			z=1<<(z-1);//转化为二进制 
			if(x>y){
     
				int t;
				t=x,x=y,y=t;
			}
			change(1);
		}
		else{
     
			scanf("%d%d",&x,&y);
			if(x>y){
     
				int t;
				t=x,x=y,y=t;
			}
			sum=0;
			ask(1);
			printf("%d\n",solve(sum));
		}
	}
	return 0;
}

你可能感兴趣的:(#,(权值)线段树,线段树染色)