【bzoj2243】染色 树链剖分+线段树

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=2243

【题解】

神坑题目，今天一天都在调这道题了。

首先asksum函数中忘记判断区间合并时出现相同颜色的情况，导致wa不断。

然后要到了数据，查出了这个错误。

然而忘记了deep[1]=1,导致在求lca的过程中访问到0号结点，然后又开始RE

然后我的一整天都在二分出错位置了。。。。。。。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
#define FILE "read"
#define MAXN 100010
#define up(i,j,n) for(int i=j;i<=n;++i)
#define dn(i,j,n) for(int i=j;i>=n;--i)
#define cmax(a,b) a=max(a,b)
#define cmin(a,b) a=min(a,b)
namespace INIT{
	char buf[1<<15],*fs,*ft;
	inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline int read(){
		int x=0,f=1;  char ch=getchar();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getchar();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getchar();}
		return x*f;
	}
}using namespace INIT;
struct node{int y,next;}e[MAXN*2];
int n,m,len,z,a[MAXN],Link[MAXN],size[MAXN],son[MAXN],deep[MAXN],f[MAXN],top[MAXN],rank[MAXN],w[MAXN],cl[MAXN*8],cr[MAXN*8],v[MAXN*8],tag[MAXN*8],anc[MAXN][25];
void insert(int x,int y){e[++len].next=Link[x];Link[x]=len;e[len].y=y;}
void updata(int p){cl[p]=cl[p<<1];cr[p]=cr[p<<1|1];v[p]=v[p<<1]+v[p<<1|1]-(cr[p<<1]==cl[p<<1|1]);}
void dfs(int x){
	size[x]=1;  son[x]=0;  anc[x][0]=f[x];
	up(i,1,20)  anc[x][i]=anc[anc[x][i-1]][i-1];
	for(int i=Link[x];i;i=e[i].next){
		if(e[i].y==f[x])  continue;
		deep[e[i].y]=deep[x]+1;  f[e[i].y]=x;
		dfs(e[i].y);  size[x]+=size[e[i].y];
		if(size[e[i].y]>size[son[x]])  son[x]=e[i].y;
	}
}
void dfs(int x,int temp){
	w[x]=++z; rank[z]=x; top[x]=temp;
	if(son[x])  dfs(son[x],temp);
	for(int i=Link[x];i;i=e[i].next)
		if(e[i].y!=f[x]&&e[i].y!=son[x])  dfs(e[i].y,e[i].y);
}
void build(int p,int l,int r){
	if(l>r)  return;
	if(l==r)  {cl[p]=cr[p]=a[rank[l]]; v[p]=1; return;}
	int mid=(l+r)>>1;
	build(p<<1,l,mid);  build(p<<1|1,mid+1,r);
	updata(p);
}
void pushdown(int p){
	if(tag[p]==-1)  return;
	int temp=tag[p]; tag[p]=-1;
	tag[p<<1]=temp;  tag[p<<1|1]=temp;
	v[p<<1]=v[p<<1|1]=1;
	cl[p<<1]=cr[p<<1]=cl[p<<1|1]=cr[p<<1|1]=temp;
}
void change(int p,int l,int r,int x,int y,int val){
	pushdown(p);
	if(x>r||y<l)  return;
	if(x<=l&&y>=r)  {tag[p]=cl[p]=cr[p]=val;v[p]=1;return;}
	int mid=(l+r)>>1;
	change(p<<1,l,mid,x,y,val);  change(p<<1|1,mid+1,r,x,y,val);
	updata(p);
}
void solvechange(int x,int y,int val){
	while(top[x]!=top[y]){
		change(1,1,z,w[top[x]],w[x],val);
		x=f[top[x]];
	}
	change(1,1,z,w[y],w[x],val);
}
int find(int p,int l,int r,int x){
	pushdown(p);
	if(x>r||x<l)  return 0;
	if(l==r)  return cl[p];
	int mid=(l+r)>>1;
	return find(p<<1,l,mid,x)+find(p<<1|1,mid+1,r,x);
}
int asksum(int p,int l,int r,int x,int y){
    pushdown(p);
    if(l==x&&r==y)return v[p];
    int mid=(l+r)>>1;
    if(mid>=y)return asksum(p<<1,l,mid,x,y);
    else if(mid<x)return asksum(p<<1|1,mid+1,r,x,y);
    else return asksum(p<<1,l,mid,x,mid)+asksum(p<<1|1,mid+1,r,mid+1,y)-(cr[p<<1]==cl[p<<1|1]);
}
int query(int x,int y){
	int ans=0;
	while(top[x]!=top[y]){
		ans+=asksum(1,1,z,w[top[x]],w[x]);
		if(find(1,1,z,w[top[x]])==find(1,1,z,w[f[top[x]]])) --ans;
		x=f[top[x]];
	}
	ans+=asksum(1,1,z,w[y],w[x]);
	return ans;
}
int lca(int x,int y){
	if(deep[x]<deep[y])  swap(x,y);
	dn(i,20,0)  if(deep[anc[x][i]]>=deep[y])  x=anc[x][i];
	if(x==y)  return x;
	dn(i,20,0)  if(anc[x][i]!=anc[y][i])  x=anc[x][i],y=anc[y][i];
	return f[x];
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	n=read();  m=read();  memset(tag,-1,sizeof(tag));
	up(i,1,n)  a[i]=read();  deep[1]=1;
	//坑点：一定要有deep[1]=1,否则会在求lca的时候出现RE，博主因此检查了一个下午
	up(i,1,n-1)  {int x=read(),y=read(); insert(x,y);insert(y,x);}
	dfs(1);  dfs(1,1);  build(1,1,z);
	//up(i,1,n)  change(1,1,z,w[i],w[i],a[i]);
	//build建树常数更小，比直接修改要快1400ms
	up(i,1,m){
		char ch[5];  scanf("%s",ch+1);
		if(ch[1]=='C') {
			int x=read(),y=read(),val=read(),t=lca(x,y);
			solvechange(x,t,val);  solvechange(y,t,val);
		}
		else {
			int x=read(),y=read(),t=lca(x,y);
			printf("%d\n",query(x,t)+query(y,t)-1);
		}
	}
	return 0;
}