TonyY是一个喜欢到处浪的男人,他的梦想是带着兰兰姐姐浪遍天朝的各个角落,不过在此之前,他需要做好规划。
现在他的手上有一份天朝地图,上面有n个城市,m条交通路径,每条交通路径都是单行道。他已经预先规划好了一些点作为旅游的起点和终点,他想选择其中一个起点和一个终点,并找出从起点到终点的一条路线亲身体验浪的过程。但是他时间有限,所以想选择耗时最小的,你能告诉他最小的耗时是多少吗?
Input包含多组测试数据。
输入第一行包括两个整数n和m,表示有n个地点,m条可行路径。点的编号为1 - n。
接下来m行每行包括三个整数i, j, cost,表示从地点i到地点j需要耗时cost。
接下来一行第一个数为S,表示可能的起点数,之后S个数,表示可能的起点。
接下来一行第一个数为E,表示可能的终点数,之后E个数,表示可能的终点。
0<S, E≤n≤100000,0<m≤100000,0<cost≤100。
输出他需要的最短耗时。
4 4 1 3 1 1 4 2 2 3 3 2 4 4 2 1 2 2 3 4Sample Output
1
如果用迪杰斯特拉算法则会爆,所以改用链式前向星算法
#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
struct Edage
{
int from;
int to;
int w;
int next;
}edage[550000];
int head[150000];
int dis[150000];
int vis[150000];
int n,m,cnt;
void add(int u,int v,int w)
{
edage[cnt].to=v;
edage[cnt].w=w;
edage[cnt].next=head[u];
head[u]=cnt++;
}
void SPFA()
{
memset(vis,0,sizeof(vis));
for(int i=0;i<=n+1;i++)
dis[i]=0x3f3f3f3f;
dis[0]=0;
queue<int >s;
s.push(0);
while(!s.empty())
{
int u=s.front();
vis[u]=0;
s.pop();
for(int i=head[u];i!=-1;i=edage[i].next)
{
int v=edage[i].to;
int w=edage[i].w;
if(dis[v]>dis[u]+w)
{
dis[v]=dis[u]+w;
if(vis[v]==0)
{
s.push(v);
vis[v]=1;
}
}
}
}
printf("%d\n",dis[n+1]);
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
cnt=0;
memset(head,-1,sizeof(head));
for(int i=0;i<m;i++)
{
int x,y,w;
scanf("%d%d%d",&x,&y,&w);
add(x,y,w);
}
int s;
scanf("%d",&s);
while(s--)
{
int x;
scanf("%d",&x);
add(0,x,0);
}
int e;
scanf("%d",&e);
while(e--)
{
int x;
scanf("%d",&x);
add(x,n+1,0);
}
SPFA();
}
}