消息传递

洛谷

树形DP
每个点跑一遍
因为你告诉一个人需要1个单位时间,所以ans+1;
可以优化(f[i][j]表示i号节点由父亲j传过来最大值。记搜)

#include<bits/stdc++.h>
#define re return
#define inc(i,l,r) for(int i=l;i<=r;++i)
const int maxn=1005;
using namespace std;
template<typename T>inline void rd(T&x)
{
	char c;bool f=0;
	while((c=getchar())<'0'||c>'9')if(c=='-')f=1;
	x=c^48;
	while((c=getchar())>='0'&&c<='9')x=x*10+(c^48);
	if(f)x=-x;
}

int k,n,hd[maxn],dis[maxn],ans[maxn],vv=999999999;

struct node{
	int to,nt;
}e[maxn<<2];

inline void add(int x,int y)
{
	e[++k].to=y;e[k].nt=hd[x];hd[x]=k;
	e[++k].to=x;e[k].nt=hd[y];hd[y]=k;
}

inline void dfs(int x,int fa)
{
	int top=0;
	int q[maxn];
        //不能全局,因为后面可能改变当前序列值
	for(int i=hd[x];i;i=e[i].nt)
	{
		int v=e[i].to;
		if(fa==v)continue;
		dfs(v,x);
		q[++top]=dis[v];
		dis[v]=0;
	} 
	
	sort(q+1,q+top+1);
	inc(i,1,top)dis[x]=max(dis[x],q[i]+top-i+1);
        //贪心,耗时多的排在前面,计算最优情况下最大时间
}

int main()
{
	int x;
	rd(n);
	inc(i,2,n)
	{
		rd(x);
		add(x,i);
	}
	
	inc(i,1,n)
	{
		if(i==8)
		x=1;
		dfs(i,0);
		ans[i]=dis[i];
		vv=min(ans[i],vv);
		dis[i]=0;
	} 
	
	printf("%d\n",vv+1);
	inc(i,1,n)
	if(ans[i]==vv)
	printf("%d ",i);
	re 0;
}
posted @ 2019-08-01 21:42  凉如水  阅读(173)  评论(0编辑  收藏  举报