【SSL1614】医院设置【最短路】

Description

　　设有一棵二叉树（如图）。其中，圈中的数字表示结点中居民的人口。圈边上数字表示结点编号，现在要求在某个结点上建立一个医院，使所有居民所走的路程之和为最小，同时约定，相邻接点之间的距离为1。如 右图中，若医院建在：
　 1处，则距离和=4+12+220+240=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

分析

我依然选用floyed大法！！
其实这题跟那棵树没啥关系，只要输入的时候注意一下左右连起来。还有最后根据题意算一下总的路程即可。

上代码

#include<iostream>
#include<cstdio>
using namespace std;
int n,x,y,v[1001],a[1001][1001],sum,ans;
int main()
{
	cin>>n;
	memset(a,0x7f/3,sizeof(a));
	for(int i=1;i<=n;i++)
	{
		cin>>v[i]>>x>>y;
		a[i][x]=1;
		a[x][i]=1;
		a[i][y]=1;
		a[y][i]=1;
		a[i][i]=0;
	}
	for(int k=1;k<=n;k++)
	{
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				a[i][j]=min(a[i][j],a[i][k]+a[k][j]);
			}
		}
	}
	ans=0x7fffffff;
	for(int i=1;i<=n;++i)
	{
		sum=0;
		for(int j=1;j<=n;++j)
		{
			sum+=v[j]*a[i][j];
		}
		ans=min(ans,sum);
	}
	cout<<ans;
	return 0; 
}