Choose the best route(Dijkstra)
原题:hdu2680
这是一道有点难度的最短路径题,要想到以最后终点为起点进行搜索,或者新建一个起点用0表示,题目给的起点与0点连接且距离为零,其他点不与0点相连接,进行搜索。我一开始用Floyd来解,发现后来怎么优化都超时,看来这种单源点题目还是要用Dijkstra来解。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n,m,s;
int ma=100000000;
int mp[1050][1050],vis[1050];
int dist[1050];
void dijkstra(int s)
{
memset(vis,0,sizeof(vis));
vis[s]=1;
for(int i=1;i<=n;i++)
{
dist[i]=mp[s][i];
}
int min_cost,min_cost_index=-1;
for(int i=2;i<=n;i++)
{
min_cost=ma;
for(int j=1;j<=n;j++)
{
if(vis[j]==0&&dist[j]<min_cost)
{
min_cost=dist[j];
min_cost_index=j;
}
}
if(min_cost_index==-1)
{
}
vis[min_cost_index]=1;
for(int j=1;j<=n;j++)
{
if(vis[j]==0&&mp[min_cost_index][j]!=ma&&mp[min_cost_index][j]+min_cost<dist[j])
{
dist[j]=mp[min_cost_index][j]+min_cost;
}
}
}
}
int main()
{
while(scanf("%d%d%d",&n,&m,&s)!=EOF)
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(i==j)
{
mp[i][j]=0;
}
else
{
mp[i][j]=ma;
}
}
}
int a,b,c;
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&a,&b,&c);
if(mp[b][a]>c)
{
mp[b][a]=c;
}
}
dijkstra(s);
int t,d,mi=ma;
scanf("%d",&t);
for(int i=0;i<t;i++)
{
scanf("%d",&d);
if(mi>dist[d])
{
mi=dist[d];
}
}
if(mi<ma)
{
printf("%d\n",mi);
}
else
{
printf("-1\n");
}
}
return 0;
}