医院设置（python）

【例3-3】医院设置

【问题描述】

     设有一棵二叉树（如图3-8，其中圈中的数字表示结点中居民的人口，圈边上数字表示结点编号。现在要求在某个结点上建立一个医院，使所有居民所走的路程之和为最小，同时约定，相邻结点之间的距离为1。就本图而言，若医院建在1处，则距离和=4+12+2*20+2*40=136；若医院建在3处，则距离和=4*2+13+20+40=81……

【输入格式】

     输入文件名为hospital.in，其中第一行一个整数n，表示树的结点数（n<=100）。接下来的n行每行描述了一个结点的状况，包含三个整数，整数之间用空格（一个或多个）分隔，其中：第一个数为居民人口数；第二个数为左链接，为0表示无链接；第三个数为右链接，为0表示无链接。

【输出格式】

     输出文件名为hospital.out，该文件只有一个整数，表示最小距离和。

【样例输入】

　　5

　　13 2 3

　　4 0 0

　　12 4 5

　　20 0 0

　　40 0 0

【样例输出】

　　81

代码部分：

import sys
s=[0]
n=int(input())
f=[[sys.maxsize for _ in range(n+1)]for _ in range(n+1)]
for i in range(1,n+1):#对每个数据进行储存
    f[i][i]=0
    x,y,z=map(int,input().split())
    s.append(x)
    if y>0:
        f[i][y]=f[y][i]=1
    if z>0:
        f[i][z]=f[z][i]=1
for k in range(1,n+1):  #最短路径Floyd算法
    for i in range(1,n+1):
        for j in range(1,n+1):
            if f[i][j]>(f[i][k]+f[k][j]):
                f[i][j]=f[i][k]+f[k][j]
s1=[]
for i in range(1,n+1):
    num=0
    for j in range(1,n+1):
        num+=f[i][j]*s[j]
    s1.append(num)
s1.sort()
print(s1[0])

思路：可以把树看成一个图，根据图的最短路径floyd算法可以得出各个点到各个点的最短距离，

根据得出的距离乘以该点的数值可以得出路径。再对数据进行排序，可以得出最短距离。