Link:http://acm.fzu.edu.cn/problem.php?pid=2261
TonyY是一个喜欢到处浪的男人,他的梦想是带着兰兰姐姐浪遍天朝的各个角落,不过在此之前,他需要做好规划。
现在他的手上有一份天朝地图,上面有n个城市,m条交通路径,每条交通路径都是单行道。他已经预先规划好了一些点作为旅游的起点和终点,他想选择其中一个起点和一个终点,并找出从起点到终点的一条路线亲身体验浪的过程。但是他时间有限,所以想选择耗时最小的,你能告诉他最小的耗时是多少吗?
包含多组测试数据。
输入第一行包括两个整数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
思路:
要确定起点和终点,不用kruskal 用spfa
另外因为起点、终点不确定,所以在所有起点前加一个0节点作为公共起点,在所有终点后加一个n+1节点作为公共终点
然后直接套模板
Code:
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<stack>
using namespace std;
const int MAXN=2e5+10;
const int INF=0x3f3f3f3f;
int n,m,f,t,v,tot,ans;
int dis[MAXN],vis[MAXN],head[MAXN];
struct edge
{
int to,w,next;
}edge[MAXN];
void build(int u,int v,int w)
{
edge[tot].to=v;
edge[tot].w=w;
edge[tot].next=head[u];
head[u]=tot++;
}
void spfa()
{
queue <int> q;
q.push(0);
vis[0]=1;
dis[0]=0;
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=0;
if(u==n+1)
{
ans=min(ans,dis[u]);
continue;
}
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].to,w=edge[i].w;
if(dis[v]>dis[u]+w)
{
dis[v]=dis[u]+w;
if(vis[v]==0)
{
q.push(v);
vis[v]=1;
}
}
}
}
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
int a,x;
tot=0;
ans=INF;
memset(dis,INF,sizeof(dis));
memset(vis,0,sizeof(vis));
memset(head,-1,sizeof(head));
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&f,&t,&v);
build(f,t,v);
}
scanf("%d",&a);
for(int i=1;i<=a;i++)
{
scanf("%d",&x);
build(0,x,0);
}
scanf("%d",&a);
for(int i=1;i<=a;i++)
{
scanf("%d",&x);
build(x,n+1,0);
}
spfa();
printf("%d\n",ans);
}
return 0;
}