【CODEVS2577】医院设置

Description

设有一棵二叉树,如下图

其中,圈中数字表示结点居民的人口.圈边上数字表示结点编号,.现在要求在某个结点上建立一个医院,使所有居民所走的路程之和为最小,同时约定,相邻结点之间 的距离为1.如上图中,若医院建在:

1处：则距离之和=4+12+2*20+2*40=136

3处：则距离之和=4*2+13+20+40=81

…….

Input

第一行一个整数n，表示树的结点数。（n<=100）

接下来的n行每行描述了一个结点的状况，包含三个整数，整数之间用空格（一个或多个）分隔，其中：第一个数为居民人口数；第二个数为左链接，为0表示表链接；第三个数为右链接。

Output

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

Sample Input

5

13 2 3

4 0 0

12 4 5

20 0 0

40 0 0

Sample Output

81

#include<iostream>
using namespace std;
int a[110][110];
int c[110];
const int MAXN=10000000;
int main()
{
    int n,l,r;
    cin>>n;
    for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) a[i][j]=MAXN;
    for (int i=1;i<=n;i++)
    {
        cin>>c[i]>>l>>r;
        if (l) a[i][l]=1,a[l][i]=1;
        if (r) a[i][r]=1,a[r][i]=1;
    }
    for (int i=1;i<=n;i++) a[i][i]=0;
    for (int k=1;k<=n;k++)
        for (int i=1;i<=n;i++)
            for (int j=1;j<=n;j++)
                if (a[i][j]>a[i][k]+a[k][j])
                    a[i][j]=a[i][k]+a[k][j];
    int ans=MAXN;
    for (int i=1;i<=n;i++)
    {
        int sum=0;
        for (int j=1;j<=n;j++)
            if (i!=j) sum+=c[j]*a[i][j];
        if (sum<ans) ans=sum;
    }
    cout<<ans;
    return 0;
}