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$。

//【參考代碼】
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N  =105; 
int map[N][N];//存圖的 
int v[N];
int n;
int main() {memset(map,0x3f,sizeof(map));scanf("%d",&n);//存儲圖 for(int i=1; i<=n; i++){int left, right;scanf("%d",&v[i]);scanf("%d",&left);scanf("%d",&right);if(left!=0){map[i][left] = 1;map[left][i] = 1;}if(right!=0){map[i][right] = 1;map[right][i] = 1;}map[i][i] = 0;}//弗洛伊德 for(int k=1; k<=n; k++){for(int i=1; i<=n; i++){for(int j=1; j<=n; j++){map[i][j] = min(map[i][j], map[i][k]+map[k][j]);}}} //醫院設置的計算式子int min = 1000;for(int i=1; i<=n; i++){int sum = 0;for(int j=1; j<=n; j++){sum+=map[i][j]*v[j];}if(sum<min){min = sum;}}cout<<min;return 0;
}