【SSL1761】城市问题【floyed最短路】

Description

设有n个城市，依次编号为0，1，2，……，n-1（n<=100），另外有一个文件保存n个城市之间的距离(每座城市之间的距离都小于等于1000)。当两城市之间的距离等于-1时，表示这两个城市没有直接连接。求指定城市k到每一个城市i（0<=I,k<=n-1）的最短距离。

Input

第一行有两个整数n和k，中间用空格隔开；以下是一个NxN的矩阵，表示城市间的距离，数据间用空格隔开。

Output

输出指定城市k到各城市间的距离（从第0座城市开始，中间用空格分开）

Sample Input

3 1
0 3 1
3 0 2
1 2 0

Sample Output

3 0 2

分析

这道题可以用floyed算法来做。
不过有一个要注意的点就是，它的点是从0开始数的，所以我在这里强行改为从1开始数的。

上代码

#include<cstdio>
#include<iostream>
using namespace std;
int n,a[101][101],m;
int main()
{
	scanf("%d%d",&n,&m);
	m++;//因为这道题的点是从0开始数，我们强行把点改为从1开始数
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			scanf("%d",&a[i][j]);
			if (a[i][j]!=-1) a[j][i]=a[i][j];
			else a[i][j]=a[j][i]=0x3fffff;
		}
	}
	
	a[m][m]=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]);//floyed算法
			}
		}
	} 
	for(int i=1;i<=n;i++)
	{
		printf("%d ",a[m][i]);
	} 
	return 0;
}