P1364 医院设置-floyd

题目描述

设有一棵二叉树，如图：

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

分析

数据规模与约定

对于 $100\%$ 的数据，保证 $1\le n\le 100，0\le u,v\le n0，1\le w\le 10^5$

---

把数据范围特意放在这里，意在告诉自己重视根据数据推算法的这一过程。
在数据量小于等于100时，可以用到的算法包括floyd，dp，高斯消元
在得到这道题考察的是多源最短路，而不是dfs遍历树，就成功把问题中的模型剥离了出来，这道题也就失去了它的神秘感

#include <bits/stdc++.h>
using namespace std;
const int N = 110;
int n;
int d[N][N], w[N];

int main() {
	cin >> n;
	
	memset(d, 0x3f, sizeof d);
	for (int i = 1; i <= n; i++) {
		int c, a, b;
		cin >> c >> a >> b;
		w[i] = c;
		d[i][i] = 1;
		if (a) d[i][a] = 1, d[a][i] = 1;
		if (b) d[i][b] = 1, d[b][i] = 1;
	}
	
	for(int k = 1; k <= n; k++) 
		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= n; j++)
				d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
		
	int ans = 1e9;
	for (int i = 1; i <= n; i++) {
		int temp = 0;
		for (int j = 1; j <= n; j++) {
			if (j == i) continue;
			temp += d[i][j] * w[j];
		}
		ans = min(ans, temp);
	}
	cout << ans << endl;
	
	return 0;
}

后话

没有想到自己的能力还是很欠缺，做不出来这种题，被降队……
知耻而后勇吧！！！
2022-03-21