P1364 医院设置(算法Floyd+树的重心转移

P1364 医院设置(算法Floyd+树的重心转移
题目传送门
题目描述

设有一棵二叉树，如图：

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

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

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

一个整数，表示最小距离和。
输入输出样例
输入 #1

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

输出 #1

81
说明/提示
数据规模与约定

对于 100% 的数据，保证 1≤n≤100，0≤u,v≤n ，1≤w≤10^5。
因为数据很水，所以O(N^3)就能过。
我就是这么水过的
用Floyd算法，可以算出两节点距离的最小值。

for(a=1;a<=n;a++){
   
        for(i=1;i<=n;i++) {
   
            for (j = 1; j <= n; j++) {
   
                if (a != i && a != j && k[a][i] + k[a][j] < k[i][j])
                    k[i][j] = k[a][i] + k[a][j];
            }
        }
    }

三重循环，得到i,j节点距离的最小值。通过不断选取比较a节点来得到。
以下为完整代码：

#include <iostream>
#include <algorithm>
#include <math.h>
#include <string>
#include <vector>
#include <map>
#include <stack>
#include <cstdio>
#include <list>
#include <queue>
#include <set>
using namespace std;
#define ll long long

int h[1000];

int k[1000][1000];
int main(){
   
    int a,j,n,i,x,y,min_1=1000000,s;
    cin>>n;
    for(i=1;i<=n;i++){
   
        for(a=1;a<=n;a++) {
   
            if(i