「SDOI2017」树点涂色

我sb的不行了

其实一开始有一个类似动态dp的想法

每个点维护到lct树上到最浅点的颜色段数，然后维护一个\(mx_{0,1}\)也就是是否用虚儿子的最大颜色

用个set维护一下虚儿子

但是啊，我发现搞这个区间改颜色的时候，虚儿子好像得用树套树维护，我当场就不行了...

每个点如果维护到根的颜色段数\(f\)

然后发现啊，这个你如果用一个lct的一个子树维护同一种颜色，在你access的时候实变虚或者虚变实对子树有一个+1或者-1

然后额外在外面开一个线段树维护子树加减，每次access的时候操作

然后这个颜色段比较特殊，链的颜色个数可以\(f_u+f_v-f_{lca(u,v)}*2+1\)，这个简单讨论一下就好了

操作3就是区间最大值

Code:

#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
using std::max;
template <class T>
void read(T &x)
{x=0;char c=getchar();while(!isdigit(c)) c=getchar();while(isdigit(c)) x=x*10+c-'0',c=getchar();
}
const int N=1e5+10;
int n,m;
int head[N],to[N<<1],Next[N<<1],cnt;
void add(int u,int v)
{to[++cnt]=v,Next[cnt]=head[u],head[u]=cnt;
}
int mx[N<<2],tag[N<<2];
int dfn[N],low[N],dep[N],yuy[N],f[20][N],dfsclock;
int LCA(int x,int y)
{if(dep[x]<dep[y]) std::swap(x,y);for(int i=18;~i;i--)if(dep[f[i][x]]>=dep[y])x=f[i][x];if(x==y) return x;for(int i=18;~i;i--)if(f[i][x]!=f[i][y])x=f[i][x],y=f[i][y];return f[0][x];
}
void pushdown(int id)
{if(tag[id]){tag[id<<1]+=tag[id];tag[id<<1|1]+=tag[id];mx[id<<1]+=tag[id];mx[id<<1|1]+=tag[id];tag[id]=0;}
}
void build(int id,int l,int r)
{if(l==r) {mx[id]=yuy[l];return;}int mid=l+r>>1;build(id<<1,l,mid),build(id<<1|1,mid+1,r);mx[id]=max(mx[id<<1],mx[id<<1|1]);
}
void modify(int id,int L,int R,int l,int r,int d)
{if(l==L&&r==R){mx[id]+=d;tag[id]+=d;return;}pushdown(id);int Mid=L+R>>1;if(r<=Mid) modify(id<<1,L,Mid,l,r,d);else if(l>Mid) modify(id<<1|1,Mid+1,R,l,r,d);else modify(id<<1,L,Mid,l,Mid,d),modify(id<<1|1,Mid+1,R,Mid+1,r,d);mx[id]=max(mx[id<<1],mx[id<<1|1]);
}
int query(int id,int l,int r,int p)
{if(l==r) return mx[id];pushdown(id);int mid=l+r>>1;if(p<=mid) return query(id<<1,l,mid,p);else return query(id<<1|1,mid+1,r,p);
}
int query(int id,int L,int R,int l,int r)
{if(L==l&&R==r) return mx[id];pushdown(id);int Mid=L+R>>1;if(r<=Mid) return query(id<<1,L,Mid,l,r);else if(l>Mid) return query(id<<1|1,Mid+1,R,l,r);else return max(query(id<<1,L,Mid,l,Mid),query(id<<1|1,Mid+1,R,Mid+1,r));
}
int ch[N][2],par[N],Lef[N];
#define fa par[now]
#define ls ch[now][0]
#define rs ch[now][1]
bool isroot(int now){return ch[fa][0]==now||ch[fa][1]==now;}
int identity(int now){return ch[fa][1]==now;}
void connect(int f,int now,int typ){ch[fa=f][typ]=now;}
void updata(int now)
{Lef[now]=now;if(ls) Lef[now]=Lef[ls];
}
void Rotate(int now)
{int typ=identity(now),p=fa;connect(p,ch[now][typ^1],typ);if(isroot(p)) connect(par[p],now,identity(p));else fa=par[p];connect(now,p,typ^1);updata(p),updata(now);
}
void splay(int now)
{for(;isroot(now);Rotate(now))if(isroot(fa))Rotate(identity(fa)^identity(now)?now:fa);
}
void access(int now)
{for(int las=0;now;las=now,now=fa){splay(now);if(rs)modify(1,1,n,dfn[Lef[rs]],low[Lef[rs]],1);if(las)modify(1,1,n,dfn[Lef[las]],low[Lef[las]],-1);rs=las;updata(now);}
}
void dfs(int now,int dis)
{dfn[now]=++dfsclock;yuy[dfsclock]=dis;dep[now]=dep[par[now]=f[0][now]]+1;for(int i=1;f[i-1][now];i++) f[i][now]=f[i-1][f[i-1][now]];for(int v,i=head[now];i;i=Next[i])if((v=to[i])!=f[0][now])f[0][v]=now,dfs(v,dis+1);low[now]=dfsclock;
}
int main()
{read(n),read(m);for(int u,v,i=1;i<n;i++) read(u),read(v),add(u,v),add(v,u);dfs(1,1);build(1,1,n);for(int op,x,y,i=1;i<=m;i++){read(op),read(x);if(op==1) access(x);else if(op==2){read(y);printf("%d\n",query(1,1,n,dfn[x])+query(1,1,n,dfn[y])-(query(1,1,n,dfn[LCA(x,y)])<<1)+1);}elseprintf("%d\n",query(1,1,n,dfn[x],low[x]));}return 0;
}

2019.4.10