题意:给你n个点树然后权值为【0,n-1】的排列,然后然后让你找出(0<=k<=n)所有k对应最大联通集合 满足mex=k;
其中mex的定义为不出现在该集合中最小的非负整数;
对于上图颜色相同的才能为同一集合,因为集合内不能有k.
对于上图假如2号点以及后面的子树的mex为k则加入1号点依然是k;
由上面俩点我们可以直接从0根点遍历(k!=0时要想mex=k则集合必须有k)
如果到达k结点,k的子树有一个小于k则没有一个集合mex=k;
如果k的子树全部大于k,则把k的子树去除剩下的点就是mex=k的最大集合;
然后当k=0时;
则找出0节点最多儿子的即可;
代码如下
# include<iostream>
# include <algorithm>
# include <vector>
const int N=1000006;
using namespace std;
int n,siz[N],mi[N],v[N];
vector<int>g[N];
int ans[N];
void dfs(int u,int fa)
{
siz[u]=1;
mi[u]=0x3f3f3f;
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(v==fa)
{
continue ;
}
dfs(v,u);
siz[u]+=siz[v];
mi[u]=min(mi[u],mi[v]);
}
if(mi[u]>u)
{
ans[u]=n-siz[u];
}
mi[u]=min(mi[u],u);
}
int main()
{
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
cin>>n;
for (int i = 1; i <= n; i++)cin >> v[i];
for (int i = 2; i <= n; i++) {
int fa; cin >> fa;
g[v[fa]].push_back(v[i]);
g[v[i]].push_back(v[fa]);
}
dfs(0, 0);
for (int i=0;i<g[0].size();i++)
{
int v=g[0][i];
ans[0] = max(ans[0], siz[v]);
}
for (int i = 0; i < n; i++)cout << ans[i] << " ";
cout << n << endl;
}