题目描述
巴蜀国的社会等级森严,除了国王之外,每个人均有且只有一个直接上级,当然国王没有上级。如果A是B的上级,B是C的上级,那么A就是C的上级。绝对不会出现这样的关系:A是B的上级,B也是A的上级。
最开始的时刻是0,你要做的就是用1单位的时间把一个消息告诉某一个人,让他们自行散布消息。在任意一个时间单位中,任何一个已经接到消息的人,都可以把消息告诉他的一个直接上级或者直接下属。
现在,你想知道:
1.到底需要多长时间,消息才能传遍整个巴蜀国的所有人?
2.要使消息在传递过程中消耗的时间最短,可供选择的人有那些?
输入格式
输入文件的第一行为一个整数N(N≤1000),表示巴蜀国人的总数,假如人按照1到n编上了号码,国王的编号是1。第2行到第N行(共N-1行),每一行一个整数,第i行的整数表示编号为i的人直接上级的编号。
输出格式
文件输出共计两行:
第一行为一个整数,表示最后一个人接到消息的最早时间。
第二行有若干个数,表示可供选择人的编号,按照编号从小到大的顺序输出,中间用空格分开。
输入输出样例
输入 #1复制
8 1 1 3 4 4 4 3
输出 #1复制
5 3 4 5 6 7
思路
是所有蓝题难度的树形dp当中较简单的。意思就是你选一个人进行消息传递,然后求最少传遍时间。
由于根不确定,所以要遍历所有点作为根。
令dp[i]为以i为子树,消息传遍节点i的子树的所有子树所需的最小时间。dp[i]取决于花费时间最多的那颗子树,再加上每1秒的传播时间,所以先告诉需要花费时间长的,再告诉短的,所以:
dp[i]=max(dp[i],dp[son]+i-1)
其中son为i的子节点,dp[son]+i-1为传递时间。
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <memory.h>
#define N 1001
#define inf 2e9+7
using namespace std;
int n,dp[N],head[N],cnt,s[N],a[N],minx(inf);
struct node
{
int nxt,to;
}e[N<<1];
inline void add(int u,int v)
{
e[++cnt].to=v;
e[cnt].nxt=head[u];
head[u]=cnt;
}
inline bool cmp(int a,int b)
{
return a>b;
}
void dfs(int u,int fa)
{
int son[N]={};
int cnt(0);
register int i;
for(i=head[u];i;i=e[i].nxt)
{
int v(e[i].to);
if(v==fa) continue;
dfs(v,u);
son[++cnt]=dp[v];//
}
sort(son+1,son+cnt+1,cmp);//先传给时间长的
for(i=1;i<=cnt;i++)
{
dp[u]=max(dp[u],son[i]+i-1);
}
dp[u]++;//别忘了1秒的传遍时间
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
register int i,j,k;
cin>>n;
for(i=2;i<=n;i++)
{
int u;
cin>>u;
add(u,i);
add(i,u);
}
for(i=1;i<=n;i++)//每个点都可以当一次根节点
{
memset(dp,0,sizeof(dp));
dfs(i,0);
minx=min(minx,dp[i]);
s[i]=dp[i];
}
cout<<minx<<endl;
for(i=1;i<=n;i++)
{
if(minx==s[i])
{
cout<<i<<' ';
}
}cout<<endl;
return 0;
}