医院设置x

医院设置x

 

题目描述 Description

设有一棵二叉树,如下图

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

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

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

…….

 

图片没弄上哈见谅见谅，codevs上有

 

 

输入描述 Input Description

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

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

输出描述 Output Description

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

样例输入 Sample Input

5

13 2 3

4 0 0

12 4 5

20 0 0

40 0 0

样例输出 Sample Output

81

数据范围及提示 Data Size & Hint

无

#include<iostream>
#include<cstring>

using namespace std;

/*
样例
5
13 2 3
4 0 0
12 4 5
20 0 0
40 0 0
*/

int n,i,j,k,x,y;

int a[1000],g[1000][1000];

/*
a数组储存第几个结点编号，
g数组储存其所对的左右孩子编号，
输入时，
若都为0，则说明无左右孩子（一个都没有），
若左为0，则说明左边无孩子，右同；
*/

int main() {
    long long t,minn=1000000;
    cin>>n;
    for(i=1; i<=n; i++) {
        cin>>a[i]>>x>>y;
        if (x!=0) g[i][x]=g[x][i]=1;//左不为0，说明左边有孩子，做标记
        if (y!=0) g[i][y]=g[y][i]=1;//右不为0，说明右边有孩子，做标记
    }
    for(i=1; i<=n; i++)
        for(j=1; j<=n; j++)
            if(g[i][j]!=1) g[i][j]=10001;//这些都无孩子
    for(k=1; k<=n; k++)
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)
                if((i!=k)&&(i!=j)&&(j!=k)&&(g[i][k]+g[k][j]<g[i][j]))
                    g[i][j]=g[i][k]+g[k][j];
    for(i=1; i<=n; i++) {
        for(j=1; j<=n; j++)
            if(i!=j) t=t+g[i][j]*a[j];
        if(t<minn) minn=t;//minn为最小
        t=0;//清0，便于下一次继续统计是否为最小
    }
    cout<<minn;
}