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
题目意思为有一个图,图有多个出发点和多个终点,问从出发点到终点所需的最小花费,此题有多个出发点和多个终点,且个数都比较大,是1e5的范围,如果枚举每个出发点和终点,肯定可以求出最佳答案,但是时间我们不能承受,所以我们可以考虑加一些边,设置一个源点为节点0,把题目所有的出发点连到该点,且权值为0,再设置一个汇点为节点n+1,把每个终点连上该节点,且权值也为0,所以就变成了求0节点到n+1节点的最短路了,我们跑一遍堆优化的迪杰斯特拉就可以了,也可以用spfa算法,我用的是dijkstra。
代码如下:
#include<stdio.h>
#include<iostream>
#include<queue>
#include<vector>
#include<string.h>
using namespace std;
const int maxn = 1e5 + 10;
struct Edg
{
int to,val;
bool operator < (const Edg& a) const
{
return val > a.val;
}
Edg(int a, int b):to(a), val(b){}
Edg(){}
};
vector<Edg>ma[maxn];
int dist[maxn], vis[maxn];
void dijkstra(int start, int n)
{
memset(vis, 0, sizeof(vis));
for(int i = 0; i <= n+1; ++ i)
{
dist[i] = maxn;
}
priority_queue<Edg>q;
dist[start] = 0;
q.push(Edg(start, 0));
while(!q.empty())
{
Edg temp = q.top();
q.pop();
int u = temp.to;
if(vis[u])
{
continue;
}
vis[u] = 1;
for(int i = 0; i < ma[u].size(); ++ i)
{
Edg res = ma[u][i];
int v = res.to;
if(!vis[v] && dist[v] > dist[u]+res.val)
{
dist[v] = dist[u]+res.val;
Edg t = Edg(v, dist[v]);
q.push(t);
}
}
}
}
int main()
{
int n, m;
while(scanf("%d%d", &n, &m) != EOF)
{
for(int i = 0; i <= n+1; ++ i)
{
ma[i].clear();
}
for(int i = 0; i < m; ++ i)
{
int from, to, val;
scanf("%d%d%d", &from, &to, &val);
Edg p = Edg(to, val);
ma[from].push_back(p);
}
int s, t;
scanf("%d", &s);
for(int i = 0; i < s; ++ i)
{
int a;
scanf("%d", &a);
Edg p = Edg(a, 0);
ma[0].push_back(p);
}
scanf("%d", &t);
for(int i = 0; i < t; ++ i)
{
int a;
scanf("%d", &a);
Edg p = Edg(n+1, 0);
ma[a].push_back(p);
}
dijkstra(0, n);
int ans = dist[n+1];
printf("%d\n", ans);
}
return 0;
}