医院设置【Dijkstra】

Time Limit:10000MS Memory Limit:65536K
Total Submit:145 Accepted:110
Case Time Limit:1000MS

---

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

---

解题思路

算法详情请见：Dijkstra

---

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
const int INF=0X7fffffff;
int n,v[500],k1,maxn,x[105],dis[500],f[500][500];
long long maxn1,ans=0X7fffffff;
void bsy(int k){
	memset(v,0,sizeof(v));
	for(int i=1;i<=n;i++)
	dis[i]=f[k][i];
	dis[k]=0;
	v[k]=1;
	for(int i=1;i<=n-1;i++)
	{
		maxn=INF;
		k1=0;
		for(int j=1;j<=n;j++)
		{
			if(!v[j]&&dis[j]<maxn)
			{
				maxn=dis[j];
				k1=j;
			}
		}
		if(k1==0) break;
		v[k1]=1;
		for(int j=1;j<=n;j++)
		{
		    if(!v[j]&&dis[k1]+1<dis[j]&&f[k1][j]==1)
			dis[j]=dis[k1]+1;
		}
	}
}
int main(){
	cin>>n;
	memset(f,0X7f,sizeof(f));
	for(int i=1;i<=n;i++) 
	{ 
		int y,z;
		cin>>x[i]>>y>>z;
		f[i][y]=f[y][i]=1;
		f[i][z]=f[z][i]=1;
	}
    for(int i=1;i<=n;i++)
    {
    	maxn1=0;
    	memset(dis,0X7f,sizeof(dis));
    	bsy(i);
    	for(int j=1;j<=n;j++)
    	if(i!=j)
	    maxn1+=x[j]*dis[j];
	    if(maxn1<ans)
	    ans=maxn1;
    }
    cout<<ans;
}

来源：CSDN

作者：SSL KJ

链接：https://blog.csdn.net/kejin2019/article/details/103996064