[SDOI2017]树点涂色

我们不难发现,维护从一个点到根的路径的颜色修改等于说是(LCT)里面的(access)操作对吧。

这时我们可以发现,一次实链修改等同于是对一个子树(+1),对一个子树(-1)

所以我们在(LCT)(access)里面修改即可。

剩下两个操作直接线段树+树链剖分即可。

代码不是特别长。

/*
  mail: mleautomaton@foxmail.com
  author: MLEAutoMaton
  This Code is made by MLEAutoMaton
*/
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<iostream>
using namespace std;
#define ll long long
#define re register
#define file(a) freopen(a".in","r",stdin);freopen(a".out","w",stdout)
inline int gi(){
	int f=1,sum=0;char ch=getchar();
	while(ch>'9' || ch<'0'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}
	return f*sum;
}
const int N=100010;
int n,m,front[N],cnt,son[N],siz[N],dep[N],fa[N],dfn[N],Time,top[N],val[N];
struct node{int to,nxt;}e[N<<1];
void Add(int u,int v){e[++cnt]=(node){v,front[u]};front[u]=cnt;}
void dfs1(int u,int ff){
	fa[u]=ff;siz[u]=1;dep[u]=dep[ff]+1;
	for(int i=front[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==ff)continue;
		dfs1(v,u);siz[u]+=siz[v];
		if(siz[son[u]]<siz[v])son[u]=v;
	}
}
void dfs2(int u,int tp){
	top[u]=tp;dfn[u]=++Time;val[Time]=u;
	if(!son[u])return;
	dfs2(son[u],tp);
	for(int i=front[u];i;i=e[i].nxt)
		if(e[i].to!=fa[u] && e[i].to!=son[u])dfs2(e[i].to,e[i].to);
}
struct tree{int val,l,r,tag;}t[N<<2];
void pushup(int o){t[o].val=max(t[o<<1].val,t[o<<1|1].val);}
void pushdown(int o){
	if(t[o].tag){
		t[o<<1].tag+=t[o].tag;t[o<<1|1].tag+=t[o].tag;
		t[o<<1].val+=t[o].tag;t[o<<1|1].val+=t[o].tag;
		t[o].tag=0;
	}
}
void build(int o,int l,int r){
	t[o].l=l;t[o].r=r;t[o].tag=0;
	if(l==r){t[o].val=dep[val[l]];return;}
	int mid=(l+r)>>1;
	build(o<<1,l,mid);build(o<<1|1,mid+1,r);
	pushup(o);
}
void modify(int o,int l,int r,int posl,int posr,int k){
	if(posl<=l && r<=posr){t[o].tag+=k;t[o].val+=k;return;}
	int mid=(l+r)>>1;pushdown(o);
	if(posl<=mid)modify(o<<1,l,mid,posl,posr,k);
	if(mid<posr)modify(o<<1|1,mid+1,r,posl,posr,k);
	pushup(o);
}
int query_val(int o,int l,int r,int pos){
	if(l==r)return t[o].val;
	int mid=(l+r)>>1;pushdown(o);
	if(pos<=mid)return query_val(o<<1,l,mid,pos);
	return query_val(o<<1|1,mid+1,r,pos);
}
int query_mx(int o,int l,int r,int posl,int posr){
	if(posl<=l && r<=posr)return t[o].val;
	int mid=(l+r)>>1,ret=0;pushdown(o);
	if(posl<=mid)ret=max(ret,query_mx(o<<1,l,mid,posl,posr));
	if(mid<posr)ret=max(ret,query_mx(o<<1|1,mid+1,r,posl,posr));
	return ret;
}
int getlca(int x,int y){
	while(top[x]^top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		x=fa[top[x]];
	}
	if(dep[x]<dep[y])swap(x,y);
	return y;
}
struct Splay{int ch[2],ff;}s[N];
bool isroot(int x){return (s[s[x].ff].ch[0]!=x)&(s[s[x].ff].ch[1]!=x);}
void rotate(int x){
	int y=s[x].ff,z=s[y].ff,k=s[y].ch[1]==x;
	if(!isroot(y))s[z].ch[s[z].ch[1]==y]=x;s[x].ff=z;
	s[y].ch[k]=s[x].ch[k^1];s[s[x].ch[k^1]].ff=y;
	s[x].ch[k^1]=y;s[y].ff=x;
}
void splay(int x){
	while(!isroot(x)){
		int y=s[x].ff,z=s[y].ff;
		if(!isroot(y))
			(s[y].ch[1]==x)^(s[z].ch[1]==y)?rotate(x):rotate(y);
		rotate(x);
	}
}
int findroot(int x){
	while(s[x].ch[0])x=s[x].ch[0];
	return x;
}
void access(int x){
	for(int y=0;x;y=x,x=s[x].ff){
		splay(x);
		if(s[x].ch[1]){
			int z=findroot(s[x].ch[1]);
			modify(1,1,n,dfn[z],dfn[z]+siz[z]-1,1);
		}
		s[x].ch[1]=y;
		if(s[x].ch[1]){
			int z=findroot(s[x].ch[1]);
			modify(1,1,n,dfn[z],dfn[z]+siz[z]-1,-1);
		}
	}
}
int main(){
#ifndef ONLINE_JUDGE
	freopen("in.in","r",stdin);
#endif
	n=gi();m=gi();
	for(int i=1;i<n;i++){int u=gi(),v=gi();Add(u,v);Add(v,u);}
	dfs1(1,1);dfs2(1,1);build(1,1,n);
	for(int i=2;i<=n;i++)s[i].ff=fa[i];
	while(m--){
		int opt=gi(),x=gi();
		if(opt==1)access(x);
		else if(opt==2){
			int y=gi();
			int lca=getlca(x,y);
			printf("%d
",query_val(1,1,n,dfn[x])+query_val(1,1,n,dfn[y])-2*query_val(1,1,n,dfn[lca])+1);
		}
		else printf("%d
",query_mx(1,1,n,dfn[x],dfn[x]+siz[x]-1));
	}
	return 0;
}
原文地址:https://www.cnblogs.com/mleautomaton/p/11418667.html