[洛谷]P2018 消息传递 (#树形dp+贪心)

题目描述

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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章