【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;
}