bzoj 2243

最新推荐文章于 2019-01-28 17:00:00 发布

qkoqhh

最新推荐文章于 2019-01-28 17:00:00 发布

阅读量174

点赞数

分类专栏：树链剖分线段树

本文链接：https://blog.csdn.net/qkoqhh/article/details/78994251

版权

线段树同时被 2 个专栏收录

17 篇文章 0 订阅

订阅专栏

树链剖分

4 篇文章 0 订阅

订阅专栏

emmm没错又是链剖。。

转化成区间之后题目的奇怪询问看出这题是非线段树不可了，记录左右端点的颜色，然后对树链进行合并。。合并时有一个麻烦的问题就是这里的区间是有方向的，所以在合并时要写多几步来处理这个方向问题。。于是代码又长了一大截。。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<cmath>
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,l,r) for(int i=l;i>=r;i--)
#define link(x) for(edge *j=h[x];j;j=j->next)
#define inf 1000000007
#define eps 1e-8
#define mem(a) memset(a,0,sizeof(a))
#define ll long long
#define succ(x) (1<<x)
#define lowbit(x) (x&(-x))
#define sqr(x) ((x)*(x))
#define ls T[i<<1]
#define rs T[i<<1|1]
#define op T[i]
#define NM 100005
#define nm 200005
#define pi 3.141592653
using namespace std;
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 f*x;
}


int top[NM],f[NM],size[NM],TOP,tot,id[NM],son[NM],d[NM];
int n,m,_x,_y,_t,a[NM],b[NM];
char _s[10];
struct edge{int t;edge*next;}e[nm],*h[NM],*o=e;
void add(int x,int y){o->t=y;o->next=h[x];h[x]=o++;}
void dfs1(int x){
	link(x)if(!f[j->t]){
		f[j->t]=x;d[j->t]=d[x]+1;
		dfs1(j->t);
		if(size[j->t]>size[son[x]])son[x]=j->t;
		size[x]+=size[j->t];
	}
	size[x]++;
}
void dfs2(int x){
	top[x]=TOP;id[x]=++tot;
	if(son[x])dfs2(son[x]);
	link(x)if(!top[j->t])dfs2(TOP=j->t);
}

struct info{
	int s,size,tag,l,r;
	info operator+(const info&o){
		if(l==-1)return o;
		if(o.r==-1)return *this;
		info f;
		f.tag=-1;f.l=l;f.r=o.r;
		f.size=size+o.size;
		f.s=s+o.s;f.s-=o.l==r;
		return f;
	}
}T[4*NM],null={0,0,-1,-1,-1};
void build(int i,int x,int y){
	int t=x+y>>1;
	if(x==y){op.s=1;op.l=op.r=b[x];op.size=1;op.tag=-1;return;}
	build(i<<1,x,t);build(i<<1|1,t+1,y);
	op=ls+rs;
}

void pushdown(int i){
	if(op.size>1&&op.tag>=0){
		ls.tag=rs.tag=op.tag;
		ls.s=rs.s=1;ls.l=ls.r=rs.l=rs.r=op.tag;
		op.tag=-1;
	}
}

void mod(int i,int x,int y){
	int t=x+y>>1;
	if(_y<x||y<_x)return;
	if(_x<=x&&y<=_y){op.s=1;op.l=op.r=op.tag=_t;return;}
	pushdown(i);
	mod(i<<1,x,t);mod(i<<1|1,t+1,y);
	op=ls+rs;
}

info query(int i,int x,int y){
	int t=x+y>>1;
	if(_y<x||y<_x)return null;
	if(_x<=x&&y<=_y)return op;
	pushdown(i);
	return query(i<<1,x,t)+query(i<<1|1,t+1,y);
}

void ch(int x,int y){
	for(;top[x]!=top[y];x=f[top[x]]){
		if(d[top[x]]<d[top[y]])swap(x,y);
		_x=id[top[x]];_y=id[x];mod(1,1,n);
	}
	if(d[x]>d[y])swap(x,y);
	_x=id[x];_y=id[y];mod(1,1,n);
}

int _query(int x,int y){
	info tx=null,ty=null;
	while(top[x]!=top[y])
		if(d[top[x]]>d[top[y]]){
			_x=id[top[x]];_y=id[x];tx=query(1,1,n)+tx;
			x=f[top[x]];
		}else{
			_x=id[top[y]];_y=id[y];ty=query(1,1,n)+ty;
			y=f[top[y]];
		}
	if(d[x]>d[y]){
		_x=id[y];_y=id[x];tx=query(1,1,n)+tx;
	}else{
		_x=id[x];_y=id[y];ty=query(1,1,n)+ty;
	}
	swap(tx.l,tx.r);
	tx=tx+ty;
	return tx.s;
}


int main(){
	freopen("data.in","r",stdin);
	n=read();m=read();
	inc(i,1,n)a[i]=read();
	inc(i,1,n-1){_x=read();_y=read();add(_x,_y);add(_y,_x);}
	dfs1(f[1]=1);dfs2(TOP=1);
	inc(i,1,n)b[id[i]]=a[i];
	build(1,1,n);
	while(m--){
		scanf("%s",_s);_x=read();_y=read();
		if(_s[0]=='Q')printf("%d\n",_query(_x,_y));
		else{_t=read();ch(_x,_y);}
	}
	return 0;
}