[树形dp]会议 洛谷P1395

题目描述

有一个村庄居住着 n 个村民，有 n−1 条路径使得这 n 个村民的家联通，每条路径的长度都为 1。现在村长希望在某个村民家中召开一场会议，村长希望所有村民到会议地点的距离之和最小，那么村长应该要把会议地点设置在哪个村民的家中，并且这个距离总和最小是多少？若有多个节点都满足条件，则选择节点编号最小的那个点。

输入格式

第一行，一个数 n，表示有 n 个村民。

接下来 n-1 行，每行两个数字 a 和 b，表示村民 a 的家和村民 b 的家之间存在一条路径。

输出格式

一行输出两个数字 x 和 y。

x 表示村长将会在哪个村民家中举办会议。

y 表示距离之和的最小值。

输入输出样例

输入 #1

4
1 2 
2 3 
3 4 

输出 #1

2 4

说明/提示

数据范围

对于 70% 数据 0^3n≤103。

对于 100% 数据 0^4n≤5×104。

题意： 在树上找一点，使各点到该点距离之和最小。

分析： 虽然给的是一棵无根树，但可以假设它是有根树，根为1号点。设dp数组含义为dp[i]表示各点到i号点的距离之和，显然dp[1]就是各点深度加和再减去n。

如图，得到了根节点的dp值后去求其子节点的dp值，设son[i]表示i结点这颗子树中结点个数，dp[2]的值相比于其父节点dp[1]的值减去了一部分，又增加了一部分。从图中可以看出减去的值是son[2]，增加的值是n-son[2]，这样可以得到状态转移方程，dp[now] = dp[pre] - 2*son[now] + n，之后统计一下最小的dp值作为答案。

实际上到各点距离加和最小的点具有特殊意义，这个点就是树的重心，这是树的重心另一个等价定义。

具体代码如下：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <string>
using namespace std;

int n, head[50005], cnt, dp[50005], dep[50005], son[50005]; 
struct edge
{
	int to, next;
}e[100005];

void add(int u, int v)
{
	e[++cnt].to = v;
	e[cnt].next = head[u];
	head[u] = cnt;
}

void dfs(int now, int pre)
{
	for(int i = head[now]; i; i = e[i].next)
		if(e[i].to != pre)
		{
			dp[e[i].to] = dp[now] - 2*(son[e[i].to]+1) + n;
			dfs(e[i].to, now);
		}
	
}

int get(int now, int pre)
{
	dep[now] = dep[pre]+1;
	int num = 0;
	for(int i = head[now]; i; i = e[i].next)
		if(e[i].to != pre)
			num += 1+get(e[i].to, now);
	son[now] = num;
	return num;
}

signed main()
{
	cin >> n;
	for(int i = 1; i < n; i++)
	{
		int u, v;
		scanf("%d%d", &u, &v);
		add(u, v), add(v, u);
	}
	get(1, 0);
	for(int i = 1; i <= n; i++)
		dp[1] += dep[i]-1;
	dfs(1, 0);
	int id = 1, _min = dp[1];
	for(int i = 2; i <= n; i++)
		if(_min > dp[i])
			_min = dp[i], id = i;
	cout << id << " " << dp[id];
    return 0;
}