loj3014 「JOI 2019 Final」独特的城市（长链剖分的妙用）

题目

https://loj.ac/problem/3014

分析

• 首先，对某一个点独特的所有城市都在他到最远点的路径上。也就是直径的某个端点。因此取一条直径，拿两个端点分别为根，做以下操作：
• 假如现在是以x为根，那么我们看看点y在x到y的路径上有哪些是独特的点。
• 容易发现，[x,y]路径上每一个点会贡献一个删除区间。然后这条路径上剩下的点就是独特的点。
• 如何快速维护这个东西？

首先，做一个长链剖分。用一个栈来维护当前到根路径上仍没有被删除的点。然后，先走长链，再走短链。
这个做法是对的，基于以下两个事实：

1. 走长链的删除区间是短的，走短链的删除区间是长的。因此，先走长链之后不需要重置栈（这保证了复杂度），在此基础上继续删除即可。
2. 已经被走过但是不在当前点到根路径上的点，其贡献的删除区间会被路径上的某个点包含。

每次需要进入儿子的时候就可以加入当前点，回父亲的时候看一下当前点如果还留着就要删掉。走到一个点x并且将他的删除区间做了之后，当前状态就是答案。
这些解决了之后，用个桶就可以随便做了。

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10;
int vis[N],n,m,c[N],re[N],ans[N];
vector<int> to[N];
int bfs(int x){
	memset(vis,0,sizeof vis);
	queue<int> q;q.push(x);
	int re=0;vis[x]=1;
	while(q.size()){
		re=q.front();q.pop();
		for(int y:to[re]){
			if(!vis[y]){
				vis[y]=1;q.push(y);
			}
		}
	}
	return re;
}

int fa[N],mx[N],er[N],dep[N];
void init(int x){
	er[x]=0;
	dep[x]=dep[fa[x]]+1;
	mx[x]=1;
	for(int y:to[x])if(y!=fa[x]){
		fa[y]=x;
		init(y);
		mx[x]=max(mx[y]+1,mx[x]);
		if(mx[y]>mx[er[x]]){
			er[x]=y;
		}
	}
}

int s[N],cnt,buf[N];
void add(int x){
	s[++*s]=x;
	cnt+=buf[c[x]]==0;
	buf[c[x]]++;
}

void deluntil(int d){
	while(*s&&dep[s[*s]]>d){
		buf[c[s[*s]]]--;
		cnt-=buf[c[s[*s]]]==0;
		(*s)--;
	}
}

void dfs(int x){
	int fm=0;
	for(int y:to[x])if(y!=fa[x]&&y!=er[x])
		fm=max(fm,mx[y]);
	deluntil(dep[x]-fm-1);
	if(er[x]) add(x),dfs(er[x]);
	deluntil(dep[x]-mx[er[x]]-1);
	ans[x]=max(ans[x],cnt);
	for(int y:to[x])if(y!=fa[x]&&y!=er[x]){
		if(s[*s]!=x)add(x);
		dfs(y);
	}
	if(s[*s]==x)deluntil(dep[x]-1);
}

void solve(int r){
	fa[r]=0;init(r);
	*s=0,cnt=0;memset(buf,0,sizeof buf);
	dfs(r);
}

int main() {
	freopen("a.in","r",stdin);
	cin >> n >> m;
	for(int i=1;i<n;i++){
		int u,v; scanf("%d %d", &u, &v);
		to[u].push_back(v),to[v].push_back(u);
	}
	for(int i=1;i<=n;i++)scanf("%d",&c[i]);
	int u=bfs(1),v=bfs(u);
	solve(u);
	solve(v);
	for(int i=1;i<=n;i++)printf("%d\n",ans[i]);
}