codevs 2577 医院设置

2577 医院设置

 

时间限制: 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

无

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=0x7f;
int ans=0x7fffffff,people[1001],map[1001][1001],l,r;
int main()
{
    int n,now=1,ans_now=0;
    memset(map,1,sizeof(map));
    cin>>n;
    for(int i=1;i<=n;++i)
    {
        cin>>people[i]>>l>>r;
        if(l!=0)
        map[l][i]=map[i][l]=1;
        if(r!=0)
        map[r][i]=map[i][r]=1;
    }
    for(int k=1;k<=n;++k)
    for(int i=1;i<=n;++i)
    for(int j=1;j<=n;++j)
    if(map[i][j]>map[i][k]+map[k][j])
    map[i][j]=map[i][k]+map[k][j];
    for(int now=1;now<=n;now++)
    {
        ans_now=0;
        for(int i=1;i<=n;++i)
        {
            if(i!=now)
            ans_now+=people[i]*map[now][i];
        }
        ans=min(ans,ans_now);
    } 
    cout<<ans<<endl;
    return 0;
}