树形dp+贪心思维 消息传递（洛谷 P2018）

消息传递

题目描述

巴蜀国的社会等级森严，除了国王之外，每个人均有且只有一个直接上级，当然国王没有上级。如果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的人直接上级的编号。

输出格式

文件输出共计两行：

第一行为一个整数，表示最后一个人接到消息的最早时间。

第二行有若干个数，表示可供选择人的编号，按照编号从小到大的顺序输出，中间用空格分开。

---

首先，这题看上去是单向边，以为是有根树，其实可以发现，如果一个点为根，那么其它与它直接相连的都是它的儿子，所以是无根树；

可以枚举每个点为根，size[i]表示以 i 为根的传递时间（注意不是最小）；

然后可以发现如果 i 的儿子的所需传递时间越大，那么越要第一个从 i 传给它，贪心思维体现；

最后ans要加1，因为传给根结点也要1单位时间；

代码：

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=1100;
const int M=2000100;
const int mod=1e9;
int n,head[N],cnt;
int dp[N],size[N];
struct Node{
	int to,nex;
}edge[N*2];
void add(int p,int q){
	edge[cnt].to=q;
	edge[cnt].nex=head[p];
	head[p]=cnt++;	
}
bool cmp(int p,int q){return p>q;}
int dfs(int p,int q){
	int size[N],tim=0,cnt=0;
	for(int i=head[p];~i;i=edge[i].nex){
		int v=edge[i].to;
		if(v!=q) size[++cnt]=dfs(v,p);
	}
	if(!cnt) return 0;
	sort(size+1,size+cnt+1,cmp);
	for(int i=1;i<=cnt;i++) tim=max(size[i]+i,tim);
	return tim;
}
int main(){
	memset(head,-1,sizeof(head));
	cin>>n;
	for(int i=2;i<=n;i++){
		int p;
		scanf("%d",&p);
		add(i,p),add(p,i);
	} 
	int ans=2e9;
	for(int i=1;i<=n;i++){//枚举根 
		dp[i]=dfs(i,-1);
		ans=min(ans,dp[i]);
	}
	cout<<ans+1<<endl;
	for(int i=1;i<=n;i++) if(dp[i]==ans) cout<<i<<" ";
	return 0;
}