codevs 2577 医院设置

Posted 2020-09-01

时间限制: 1 s

 空间限制: 32000 KB

 题目等级 : 黄金 Gold

题目描述 Description

设有一棵二叉树,如下图

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

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

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

…….

输入描述 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

无

n<=100 Floyd

屠龙宝刀点击就送

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>

#define MAXN 101
#define inf 0x7fffffff
using namespace std;

int ans=inf,n,sum[MAXN],i,j,atlas[MAXN][MAXN],tot,lef,righ;
void floyd()
{
    for(int i=1;i<=n;++i)
    {
        for(int j=1;j<=n;++j)
        {
            for(int k=1;k<=n;++k)
            {
                if(atlas[j][k]>atlas[j][i]+atlas[i][k])
                atlas[j][k]=atlas[j][i]+atlas[i][k];
            }
        }
    }
}
int main()
{
    memset(atlas,1,sizeof(atlas));
    cin>>n;
    for(i=1;i<=n;++i)
    {
        cin>>sum[i]>>lef>>righ;
        if(lef)
        atlas[lef][i]=atlas[i][lef]=1;
        if(righ)
        atlas[righ][i]=atlas[i][righ]=1;
    }
    floyd();
    int fart,pos=1;
    while(pos!=n+1)
    {
        fart=0;
        for(i=1;i<=n;++i)
        if(i!=pos)
        fart+=(sum[i])*atlas[pos][i];
        pos++;
        ans=min(ans,fart);
    }
    cout<<ans<<endl;
    return 0;
}