百度地图导航 (最短路)
百度地图导航
题意
给出 n 个城市, m 个城市群,每个城市群包含多个城市(一个城市可能被包含在多个城市群中),现在要修路,在城市与城市之间修路,或在城市群间修路,如果在城市群间修路,等价于两个城市群的城市两两各修一条路,问从城市 s 到 t 的最短路。
分析
问题主要是处理城市群间的相连。先考虑将每个城市群抽象成一个整体,再考虑将整体分成两个点,对于一个城市群 a ,我们新增一个入口点 a1,一个出口点 a2,那么从 a1 连边到城市群里的所有点,距离为0,从城市群里所有点连边到 a2,距离为0 。对于另一个城市群 b ,把 b2 连边 a1,a2 连边 b1,花费为城市群间的花费,等价于这两个城市群连接起来了。
code
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, int> P;
const int MAXN = 2e5 + 10;
const ll INF = 1e17;
int n, m;
struct edge {
int to, cost;
};
vector<edge> G[MAXN];
vector<pair<int, int> > groupid;
ll d[MAXN];
void dijkstra(int s)
{
priority_queue<P, vector<P>, greater<P> > que;
fill(d, d + MAXN, INF);
d[s] = 0;
que.push(P(0, s));
while(!que.empty())
{
P p = que.top();
que.pop();
int v = p.second;
if(d[v] < p.first) continue;
for(int i = 0; i < G[v].size(); i++)
{
edge e = G[v][i];
if(d[e.to] > d[v] + e.cost)
{
d[e.to] = d[v] + e.cost;
que.push(P(d[e.to], e.to));
}
}
}
}
int main() {
cin >> n >> m;
for(int i = 1; i <= m; i++) {
int k;
cin >> k;
groupid.push_back(pair<int, int>(n + 1, n + 2));
for(int j = 0; j < k; j++) {
int x;
cin >> x;
G[n + 1].push_back(edge{x, 0});
G[x].push_back(edge{n + 2, 0});
}
n += 2;
}
int m1;
cin >> m1;
for(int i = 0; i < m1; i++) {
int u, v, c;
cin >> u >> v >> c;
G[u].push_back(edge{v, c});
G[v].push_back(edge{u, c});
}
int m2;
cin >> m2;
for(int i = 0; i < m2; i++) {
int a, b, l;
cin >> a >> b >> l;
a--; b--;
G[groupid[a].second].push_back(edge{groupid[b].first, l});
G[groupid[b].second].push_back(edge{groupid[a].first, l});
}
int s, t;
cin >> s >> t;
dijkstra(s);
if(d[t] == INF) d[t] = -1;
cout << d[t] << endl;
return 0;
}