【SSL】1761城市问题

【SSL】1761城市问题

Time Limit:1000MS
Memory Limit:65536K

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

思路

用dij.

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
struct jgt
{
	int w,s;
};
struct NOTE
{
	int x,y,nxt,s;
}f[10010];
bool operator < (const jgt &a,const jgt &b)
{
	return a.s>b.s;
}
priority_queue<jgt>q;
int e[110][110],dis[110],head[110],tot;
bool d[110];
void DIJ(int s,int n)
{
	jgt tem,temp;
	int i,j,mn,w,v;
	memset(dis,100,sizeof(dis));
	memset(d,0,sizeof(d));
	for(i=1;i<=n;i++)
	{
		e[i][i]=0;
		dis[i]=e[s][i];
	}
	for(i=head[s];i>0;i=f[i].nxt)
	{
		tem.w=f[i].y;
		tem.s=dis[f[i].y];
		q.push(tem);
	}
	for(dis[s]=0,d[s]=1;!q.empty();)//堆优化 
	{
		tem=q.top();
		q.pop();
		d[tem.w]=1;
		for(j=head[tem.w];j;j=f[j].nxt)
		{
			if(!d[f[j].y]&&tem.s+e[tem.w][f[j].y]<dis[f[j].y])//松弛 
			{
				dis[f[j].y]=tem.s+e[tem.w][f[j].y];
				temp.w=f[j].y;
				temp.s=dis[f[j].y];
				q.push(temp);
			}
		}
	}
	return;
}
void add(int u,int v,int w)
{
	f[++tot].x=u;
	f[tot].y=v;
	f[tot].s=w;
	f[tot].nxt=head[u];
	head[u]=tot;
	return;
}
int main()
{
	int n,m,i,j,w;
	memset(e,100,sizeof(e));
	scanf("%d%d",&n,&m);
	tot=0;
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=n;j++)
		{
			scanf("%d",&w);
			if(w>-1)
			{
				add(i,j,w);//建边 
				e[i][j]=w;	
			}
		}
	}
	DIJ(m+1,n);
	for(i=1;i<=n;i++)
		printf("%d ",dis[i]);
	return 0;
}