百度地图上有 n 个城市,城市编号依次为 1 到 n。地图中有若干个城市群,编号依次为 1 到 m。每个城市群包含一个或多个城市;每个城市可能属于多个城市群,也可能不属于任何城市群。
地图中有两类道路。第一类道路是 城市之间的快速路,两个城市 u,v 之间增加一条距离为 c 的边;第二类道路是 城市群之间的高速路,连接两个城市群 a,b,通过这条高速路,城市群 a 里的每个城市与城市群 b 里的每个城市之间两两增加一条距离为 c 的边。图中所有边均为无向边。
你需要计算从城市 s 到城市 t 的最短路。
输入格式
第一行输入 n(1≤n≤20000), m(0≤m≤20000),分别表示城市总数和城市群总数。
接下来一共输入 m 行。
第 i 行首先输入一个 ki(1≤ki≤n),表示第 i 个城市群中的城市数为 ki。接下来输入 ki 个数,表示第 i 个城市群中每个城市的编号(保证一个城市群内的城市编号不重复且合法,∑i=1mki≤20000)。
下一行输入一个整数 m1(0≤m1≤20000),表示有 m1 条第一类道路,即 城市之间的快速路。
接下来 m1 行,每行输入三个整数 ui,vi(1≤ui,vi≤n),ci(1≤ci≤106),分别表示快速路连接的两个城市编号和边的距离。
下一行输入一个整数 m2(0≤m2≤20000),表示有 m2 条第二类道路,即 城市群之间的高速路。
接下来 m2 行,每行输入三个整数 ai,bi(1≤ai,bi≤m),li(1≤li≤106),分别表示快速路连接的两个城市群编号和边的距离。
最后一行输入 s,t(1≤s,t≤n),表示起点和终点城市编号。
输出格式
输出一个整数,表示城市 s 到城市 t 到最短路。如果不存在路径,则输出-1
。
样例说明
1 -> 2 - > 5
或者1 -> 4 -> 5
是最短的路径,总长度为 12。
样例输入
5 4 2 5 1 2 2 4 1 3 2 3 4 2 1 2 9 1 5 18 2 1 2 6 1 3 10 1 5
样例输出
12
把每个城市群抽象成两个点 s′,s′′。按照如下方式见图:
对于每个城市群里面的城市 si,连边:(si,s′,0), (s′′,si,0);
第一种边正常连边:(u,v,c),(v,u,c);
第二种边连边:(a′,b′′,l), (b′,a′′,l)。
然后跑一遍 s−t 最短路就是答案。
#include <iostream> #include <queue> #include <algorithm> #include <string.h> #include <stdio.h> using namespace std; const int maxn = 100005; const long long INF = 1e15; int n, m, m1; int visited[maxn]; long long dis[maxn]; vector<pair<int, long long>> map[maxn]; void add(int u, int v, int w) { map[u].push_back({v, w}); } void spfa(int start) { for (int i = 1; i <= n + m*2; ++i) dis[i] = INF; visited[start] = 1; dis[start] = 0; queue<int> que; que.push(start); while (!que.empty()) { int cur = que.front(); que.pop(); visited[cur] = 0; for (int i = 0; i < map[cur].size(); ++i) { if (dis[map[cur][i].first] > map[cur][i].second + dis[cur]) { dis[map[cur][i].first] = map[cur][i].second + dis[cur]; if (!visited[map[cur][i].first]) { visited[map[cur][i].first] = 1; que.push(map[cur][i].first); } } } } } int main() { cin >> n >> m; for (int i = 1; i <= m; ++i) { int k, u; cin >> k; while (k--) { cin >> u; add(u, n + i, 0); add(n + m + i, u, 0); } } cin >> m1; for (int i = 1; i <= m1; ++i) { int u, v, w; cin >> u >> v >> w; add(u, v, w); add(v, u, w); } cin >> m1; for (int i = 1; i <= m1; ++i) { int u, v, w; cin >> u >> v >> w; add(u + n, v + n + m, w); add(v + n, u + n + m, w); } int start, end; cin >> start >> end; spfa(start); if (dis[end] == INF) cout << -1 << endl; else cout << dis[end] << endl; }