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

13 2 3

4 0 0

12 4 5

20 0 0

40 0 0

样例输出 Sample Output

数据范围及提示 Data Size & Hint

无

思路：spfa+暴力

 1 #include<cstdio>
 2 #include<cstring>
 3 
 4 int map[101][101];
 5 int peo[101];
 6 int dis[101];
 7 int team[101],head,tail;
 8 bool vis[101];
 9 int n,sum,ans,minn=9999999;
10 void juli(int a)
11 {
12     memset(dis,127,sizeof(dis));
13     memset(vis,0,sizeof(vis));
14     memset(team,0,sizeof(team));
15     tail=0;
16     head=0;
17     dis[a]=0;
18     vis[a]=1;
19     team[tail++]=a;
20     while(head<tail)
21     {
22         int d=team[head];
23         head++;
24         for(int i=1;i<=n;++i)
25         {
26             if(map[d][i]&&dis[i]>dis[d]+map[d][i])
27             {
28                 dis[i]=map[d][i]+dis[d];
29                 if(!vis[i])
30                 {
31                     team[tail++]=i;
32                     vis[i]=1;
33                 }
34             }
35         }
36         vis[d]=0;
37     }
38     for(int i=1;i<=n;++i)
39     {
40         sum+=peo[i]*dis[i];
41     }
42 }
43 int main()
44 {
45     scanf("%d",&n);
46     for(int i=1;i<=n;++i)
47     {
48         int a,b,c;
49         scanf("%d%d%d",&c,&a,&b);
50         if(a)
51         {
52             map[i][a]=1;
53             map[a][i]=1;
54         }    
55         if(b)
56         {
57             map[b][i]=1;
58             map[i][b]=1;
59         }
60         peo[i]=c;
61     }
62     for(int i=1;i<=n;++i)
63     {
64         sum=0;
65         juli(i);
66         if(sum<minn)
67         {
68             minn=sum;
69         }
70     }
71     printf("%d",minn);
72     return 0;
73 }