HPU组队选拔赛(三)FZU - 2261【最短路】

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。

Output

输出他需要的最短耗时。

Sample Input
4 4
1 3 1
1 4 2
2 3 3
2 4 4
2 1 2
2 3 4
Sample Output
1

最短路模板题。用spfa邻接表优化写的。一下是AC代码:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#define INF 0x3f3f3f3f
using namespace std;
int vis[100005],dis[100005];
//int cost[1005][1005];
int n,m,top;
int head[100005];
struct node{
	int to,next,cost;
}e[100005];
void addEdge(int u,int v,int cot){
	e[top].to=v;
	e[top].next=head[u];
	e[top].cost=cot;
	head[u]=top++;
} 
void spfa(int s){
	//memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++){
		dis[i]=INF;
		vis[i]=0;
	}
	queue<int> que;
	vis[s]=1;
	dis[s]=0;
	que.push(s);
	while(!que.empty()){
		int x=que.front();
		que.pop();
		vis[x]=0;
		for(int i=head[x];i!=-1;i=e[i].next){
			int y=e[i].to;
			int z=e[i].cost;
			if(dis[y]>dis[x]+z){
				dis[y]=dis[x]+z;
				if(!vis[y]){
					que.push(y);
					vis[y]=1;
				}
			}
		}
	} 
}
int main(){
	while(~scanf("%d%d",&n,&m)){
		top=1;
		memset(head,-1,sizeof(head));
		//memset(cost,0,sizeof(cost));
		int a,b,cost;
		int sta[100005],ed[100005];
		while(m--){
			scanf("%d%d%d",&a,&b,&cost);
			addEdge(a,b,cost);
		}
		int s,e;
		scanf("%d",&s);
		for(int i=1;i<=s;i++){
			scanf("%d",&sta[i]);
		}
		scanf("%d",&e);
		for(int i=1;i<=e;i++){
			scanf("%d",&ed[i]);
		}
		int minn=INF;
		for(int i=1;i<=s;i++){
			spfa(sta[i]);
			for(int j=1;j<=e;j++){
				minn=min(minn,dis[ed[j]]);
			}
		}
		printf("%d\n",minn);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/xiang_hehe/article/details/80266896
今日推荐