
Accept: 300 Submit: 940
Time Limit: 1500 mSec Memory Limit : 32768 KB
Problem Description
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
Sample Output
Source
福州大学第十四届程序设计竞赛_重现赛
思路:这题通过简单的变换就可以变为一道最短路模板题
我们设置一个总起点,这个总起点到所有规划起点的距离为0。
我们再设置一个总终点,这个总终点到所有规划终点的距离为0。
这时我们再求总起点到总终点的最短路即可
这里用到了spaf算法求最短路
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
#define LL long long
#define M(a,b) memset(a,b,sizeof(a))
const int MAXN = 1e5+5;
const int INF = 0x3f3f3f3f;
struct Node{
int from;
int to;
int val;
int next;
}edge[MAXN];
int head[MAXN];
int dist[MAXN];
int vis[MAXN];
int n,m;
int st,ed,way;
int n1,n2,st1,ed1;
int cnt;
queue<int>q;
void add(int from,int to,int val)///建图
{
edge[cnt].to = to;
edge[cnt].val = val;
edge[cnt].next = head[from];
head[from] = cnt++;
}
void SPAF()
{
M(vis,0);
fill(dist,dist+n+2,INF);
dist[0] = 0;
while(!q.empty()) q.pop();
q.push(0);
while(!q.empty())
{
int temp = q.front();
vis[temp] = 0;
q.pop();
for(int i=head[temp];i!=-1;i=edge[i].next)
{
int temp2 = edge[i].to;
int temp3 = edge[i].val;
if(dist[temp2]>dist[temp]+temp3)
{
dist[temp2] = dist[temp] + temp3;
if(vis[temp2] == 0)
{
q.push(temp2);
vis[temp2] = 1;
}
}
// printf("---\n");
}
}
}
int main()
{
while(~scanf("%d %d",&n,&m))
{
cnt = 0;
M(head,-1);
for(int i=0;i<m;i++)
{
scanf("%d %d %d",&st,&ed,&way);
add(st,ed,way);
}
scanf("%d",&n1);
for(int i =0;i<n1;i++)///设置总起点,总起点为0
{
scanf("%d",&st1);
add(0,st1,0);
}
scanf("%d",&n2);
for(int i =0;i<n2;i++)///设置总终点,总终点为n+1
{
scanf("%d",&ed1);
add(ed1,n+1,0);
}
SPAF();
printf("%d\n",dist[n+1]);
}
return 0;
}