警钟长鸣!!!
对于这道数据很水的题,蒟蒻一头栽进了一个大坑:
DFS跑图的时候,对于每一次dfs(i),更新其子节点的时候用数组son来记录期间产生的所有传播时间,我为了省空间把son数组开在了外面,并且认为反正是要反复清空为0的就不用管,但是没有问题的只有向下传递的过程,在回溯的时候如果是全局,就会发现记下的u的son数组在计算v的时候被清空了
感谢@stdios @Sprague_Garundy 指正orz
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1010;
int n,tot,head[N],_clock=2e9,cnt,f[N],ans[N];
bool vis[N];
struct edge{int to,next;}e[N<<1];
int read()
{
int x=0,f=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void add_edge(int u,int v)
{
e[++tot].to=v;
e[tot].next=head[u];
head[u]=tot;
}
void dfs(int u)
{
int son[1010],num=0;
vis[u]=1;num=0;
memset(son,0,sizeof son);
// for(int i=1;i<=n;i++) printf("%d ",son[i]);
for(int i=head[u];i;i=e[i].next)
{
int v=e[i].to;
if(vis[v]) continue;
dfs(v);
son[++num]=f[v];
}
sort(son+1,son+num+1);
for(int i=1;i<=num;i++) f[u]=max(f[u],son[i]+num-i+1);
}
signed main()
{
n=read();
for(int i=1;i<n;i++)
{
int fa=read();
add_edge(i+1,fa);
add_edge(fa,i+1);
}
for(int i=1;i<=n;i++)
{
memset(f,0,sizeof f);
memset(vis,0,sizeof vis);
dfs(i);
if(_clock>f[i])
{
_clock=f[i];
cnt=0;
ans[++cnt]=i;
}
else if(_clock==f[i]) ans[++cnt]=i;
}
printf("%lld\n",_clock+1);
for(int i=1;i<=cnt;i++) printf("%lld ",ans[i]);
return 0;
}
/*
8
1
1
3
4
4
4
3
*/