数列分块入门8(区间修改)

原题:http://www.caioj.cn/problem.php?id=1251

题解：分块，v[]为一整块颜色统一的颜色，若v==-1 则这个块的颜色不一样需要暴力修改。

#include<bits/stdc++.h>
#define reg register

using namespace std;
const int N=110000,M=550;
int a[N],pos[N],v[M],n,m;

inline int rd(){
	int x=0;int f=1;char s=getchar();
	while(!isdigit(s)) f=(f=='-'?-1:f),s=getchar();
	while(isdigit(s)) x=(x<<1)+(x<<3)+s-'0',s=getchar();
	return x*f;
}
inline void pushdown(int x){
	if(v[x]==-1) return ;
	for(reg int i=(x-1)*m+1;i<=min(n,x*m);i++) a[i]=v[x];
	v[x]=-1;	
}
inline int query(int l,int r,int c){
	int ans=0;
	pushdown(pos[l]);
	for(reg int i=l;i<=min(r,pos[l]*m);i++)	{
		if(a[i]==c) ans++;
		else a[i]=c; 
	}
	if(pos[l]!=pos[r]){
		pushdown(pos[r]);
		for(reg int i=(pos[r]-1)*m+1;i<=r;i++) {
			if(a[i]==c) ans++;
			else a[i]=c;
		}
	}
	for(reg int i=pos[l]+1;i<=pos[r]-1;i++){
		if(v[i]==-1){
			for(int j=(i-1)*m+1;j<=i*m;j++)
				if(a[j]==c) ans++;
		}
		if(v[i]==c) ans+=m;
		v[i]=c;
	}
	return ans;
}
int main(){
//	freopen("input.in","r",stdin);
	n=rd();m=sqrt(n);
	for(reg int i=1;i<=n;i++) a[i]=rd(),pos[i]=(i-1)/m+1;
	for(reg int i=1;i<=pos[n];i++) v[i]=-1;
	for(reg int i=1,l,r,c;i<=n;i++){
		l=rd();r=rd();c=rd();
		printf("%d\n",query(l,r,c));
	}
	return 0;
}