洛谷 P1364医院设置

医院设置

题目描述

设有一棵二叉树，如图：

其中，圈中的数字表示结点中居民的人口。圈边上数字表示结点编号，现在要求在某个结点上建立一个医院，使所有居民所走的路程之和为最小，同时约定，相邻接点之间的距离为 $1$。如上图中，若医院建在 $1$ 处，则距离和 $=4+12+2\times 20+2\times 40=136$；若医院建在 $3$ 处，则距离和 $=4\times 2+13+20+40=81$。

输入格式

第一行一个整数 $n$，表示树的结点数。

接下来的 $n$ 行每行描述了一个结点的状况，包含三个整数 $w,u,v$，其中 $w$ 为居民人口数，$u$ 为左链接（为 $0$ 表示无链接），$v$ 为右链接（为 $0$ 表示无链接）。

输出格式

一个整数，表示最小距离和。

样例 #1

样例输入 #1

5						
13 2 3
4 0 0
12 4 5
20 0 0
40 0 0

样例输出 #1

81

提示

数据规模与约定

对于 $100\%$ 的数据，保证 $1\le n\le 100$，$0\le u,v\le n$，$1\le w\le 10^5$。
很好的题解使我的编译器旋转！
https://www.luogu.com.cn/article/glqqp5lh

#include <bits/stdc++.h>
using namespace std;
int dp[105] = {0},size[105] = {0},w[105] = {0};
vector<int> to[300];
int dfs(int now,int fa,int dep)
{
	for(int i=0;i<to[now].size();i++)
	{
		if(to[now][i]!= fa)
		size[now]+=dfs(to[now][i],now,dep+1);
	}
	dp[1] +=w[now]*dep;
	return size[now];
}
int ans = INT_MAX;
void solve(int now,int fa)
{
	for(int i=0;i<to[now].size();i++)
	{
		if(to[now][i]!= fa)
		{
			dp[to[now][i]] = dp[now]+size[1]-2*size[to[now][i]];
		solve(to[now][i],now);
		}
		
	}
	ans = min(ans,dp[now]);
}
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>w[i];
		size[i] = w[i];
		int l,r;
		cin>>l>>r;
		if(l!=0) 
		{
			to[i].push_back(l);
			to[l].push_back(i);
		}
		if(r!= 0 )
		{
			to[i].push_back(r);
			to[r].push_back(i);
		}
	}
	dfs(1,0,0);
	solve(1,0);
//	for(int i =1;i<=n;i++)
//	{
//		cout<<i<<" "<<size[i]<<endl;
//	}
//	cout<<endl<<dp[1]<<endl;
	cout<<ans;
	return 0;
}