【SSL 1761】城市問題【最短路 Dijkstra】

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

分析&說明：

這道題是最短路問題，選用的是Dijkstra，其中大部分是模板，開頭賦初值時注意一下-1，還有就是所有循環都要從0開始，最後依次輸出c數組就完事了。

代碼：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,k,f[101][101],c[101],p,minn;
bool u[101];
int main()
{
	cin>>n>>k;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			cin>>f[i][j];
			if(f[i][j]==-1) f[i][j]=1e30;  //賦最大值
			f[j][i]=f[i][j];
		}
	}
	memset(u,0,sizeof(u));
	u[k]=1;
	c[k]=0;
	for(int i=0;i<n;i++)
	{
		c[i]=f[k][i];
	}
	for(int i=0;i<n-1;i++)
	{
	//Dijkstra主體部分
		minn=0x7fffffff;
		p=0;
		for(int j=0;j<n;j++)
		if(!u[j]&&c[j]<minn)
		{
			minn=c[j];
			p=j;
		}
		if(p==-1) break;
		u[p]=1;
		for(int j=0;j<n;j++)
		if(c[p]+f[p][j]<c[j]&&!u[p])
		c[j]=c[p]+f[p][j];
	}
	for(int i=0;i<n;i++)
	{
		cout<<c[i]<<" ";
		//依次輸出最短路
	}
	return 0;
}