Description
给出一张有向图,n个顶点m条边(有重边),每条边带边权,给出终点以及一系列起点,问是否存在一条从起点到终点的最短路,如有,输出最短路,否则输出-1
Input
多组输入,每组用例第一行为三个整数n,m,end表示点数,边数和终点编号,之后m行每行三个整数u,v和c表示从u点到v点存在一条权值为c的边,之后为一整数q表示起点数量,最后一行为q个整数,每个数都表示一个起点编号,以文件尾结束输入
Output
对于每组用例,若存在从起点到终点的最短路则输出,否则输出-1
Sample Input
5 8 5
1 2 2
1 5 3
1 3 4
2 4 7
2 5 6
2 3 5
3 5 1
4 5 1
2
2 3
4 3 4
1 2 3
1 3 4
2 3 2
1
1
Sample Output
1
-1
Solution
构造虚拟起点0,从0到各起点之间建权值为0的边,之后从0开始用dijkstra跑一遍最短路判断dis[end]是否为INF即可
Code
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<stack>
#include<vector>
#include<string>
using namespace std;
#define maxn 1111
#define INF 0x3f3f3f3f
typedef long long ll;
int n,m,map[maxn][maxn],dis[maxn],vis[maxn];
void Dijkstra(int s)
{
for(int i=0;i<=n;i++)
{
dis[i]=map[s][i];
vis[i]=0;
}
vis[s]=1;
for(int i=1;i<=n;i++)
{
int pos,temp=INF;
for(int j=1;j<=n;j++)
if(!vis[j]&&temp>dis[j])
temp=dis[j],pos=j;
if(temp==INF)break;
vis[pos]=1;
for(int j=1;j<=n;j++)
if(!vis[j]&&dis[j]>dis[pos]+map[pos][j])
dis[j]=dis[pos]+map[pos][j];
}
}
int main()
{
int end,q,start;
while(~scanf("%d%d%d",&n,&m,&end))
{
for(int i=0;i<=n;i++)//初始化
for(int j=0;j<=n;j++)
map[i][j]=(i==j?0:INF);
while(m--)
{
int u,v,c;
scanf("%d%d%d",&u,&v,&c);
map[u][v]=min(map[u][v],c);//建边,注意重边
}
scanf("%d",&q);
while(q--)
{
scanf("%d",&start);
map[0][start]=0;//0到各起点建权值为0的边
}
Dijkstra(0);//最短路
if(dis[end]==INF)printf("-1\n");//不存在最短路
else printf("%d\n",dis[end]);//存在最短路
}
return 0;
}