【LOJ6284】数列分块8

题目大意：维护一个序列，支持区间染色，查询区间中等于某个颜色的点的个数。

题解：考虑直接用线段树进行维护，维护区间相同颜色值和一个区间颜色是否相同的标记即可。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;

int n,m,a[maxn];
struct node{
	#define ls(x) t[x].lc
	#define rs(x) t[x].rc
	int lc,rc,val,all;
}t[maxn<<1];
int tot,root;
inline void pushup(int o){
	if(t[ls(o)].val==t[rs(o)].val&&t[ls(o)].all&&t[rs(o)].all)t[o].val=t[ls(o)].val,t[o].all=1;
	else t[o].all=0;
}
inline void pushdown(int o){
	if(t[o].all){
		t[ls(o)].all=t[rs(o)].all=1;
		t[ls(o)].val=t[rs(o)].val=t[o].val;
	}
}
int build(int l,int r){
	int o=++tot;
	if(l==r){t[o].val=a[l],t[o].all=1;return o;}
	int mid=l+r>>1;
	ls(o)=build(l,mid),rs(o)=build(mid+1,r);
	return pushup(o),o;
}
int query(int o,int l,int r,int x,int y,int c){
	if(l==x&&r==y&&t[o].all){
		if(t[o].val==c)return r-l+1;
		else return t[o].val=c,0;
	}
	int mid=l+r>>1;
	pushdown(o);
	int ans=0;
	if(y<=mid)ans=query(ls(o),l,mid,x,y,c);
	else if(x>mid)ans=query(rs(o),mid+1,r,x,y,c);
	else ans=query(ls(o),l,mid,x,mid,c)+query(rs(o),mid+1,r,mid+1,y,c);
	return pushup(o),ans;
}

int main(){
	scanf("%d",&n),m=n;
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	root=build(1,n);
	while(m--){
		int l,r,c;
		scanf("%d%d%d",&l,&r,&c);
		printf("%d\n",query(root,1,n,l,r,c));
	}
	return 0;
}