题目
Description
Alice和Bob现在要乘飞机旅行,他们选择了一家相对便宜的航空公司。该航空公司一共在n个城市设有业务,设这些城市分别标记为0到n-1,一共有m种航线,每种航线连接两个城市,并且航线有一定的价格。Alice和Bob现在要从一个城市沿着航线到达另一个城市,途中可以进行转机。航空公司对他们这次旅行也推出优惠,他们可以免费在最多k种航线上搭乘飞机。那么Alice和Bob这次出行最少花费多少?
Input
数据的第一行有三个整数,n,m,k,分别表示城市数,航线数和免费乘坐次数。
第二行有两个整数,s,t,分别表示他们出行的起点城市编号和终点城市编号。(0<=s,t<n)
接下来有m行,每行三个整数,a,b,c,表示存在一种航线,能从城市a到达城市b,或从城市b到达城市a,价格为c。(0<=a,b<n,a与b不相等,0<=c<=1000)
Output
只有一行,包含一个整数,为最少花费。
Sample Input
5 6 1
0 4
0 1 5
1 2 5
2 3 5
3 4 5
2 3 3
0 2 100
Sample Output
8
HINT
对于30%的数据,2<=n<=50,1<=m<=300,k=0;
对于50%的数据,2<=n<=600,1<=m<=6000,0<=k<=1;
对于100%的数据,2<=n<=10000,1<=m<=50000,0<=k<=10.
题解
这是一个分层图的模板题。
把图分成k层,每层仍然是n个点,每层的连边方式仍然和原图相同。除此以外,层与层之间该如何连边呢?
我们假设现在在第i层,点u和点v之间有一条权值w的边,对于下一层,即i+1层,u’与v’之间仍是有一条权为w的边,且对于u与v’之间应该加上一条边权为0的边。所以如果每一层有m条边,第i层和第i+1层之间也一定是有m条边。
对于我们当前找到的终点,尝试起点的状态去更新,不选择此条边免费的状态和选择此条边免费的状态,再将这两个状态压入队列去更新可以到达的其他状态。
CODE
#include <algorithm>
#include <cctype>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <deque>
#include <functional>
#include <list>
#include <map>
#include <iomanip>
#include <iostream>
#include <set>
#include <queue>
#include <stack>
#include <string>
#include <vector>
using namespace std;
const int maxn = 2e6 + 100;
typedef long long ull;
typedef pair <int, int > pii;
inline int read() {
int s = 0, w = 1;
char ch = getchar();
while (!isdigit(ch)) { if (ch == '-') w = -1; ch = getchar(); }
while (isdigit(ch)) { s = (s << 1) + (s << 3) + (ch ^ 48); ch = getchar(); }
return s * w;
}
int n, m, s, t, k, num_edge = 0, ans = 0x3f3f3f3f;
struct Edge { int next, to, dis; } edge[maxn << 1];
int head[maxn], dis[maxn], vis[maxn];
priority_queue <pii, vector <pii>, greater<pii> > q;
inline void add(int from, int to, int dis) {
edge[++num_edge].to = to;
edge[num_edge].dis = dis;
edge[num_edge].next = head[from];
head[from] = num_edge;
}
inline int point(int x, int y) { return x + y * n; }
void dijkstra(int k) {
memset(dis, 0x3f, sizeof(dis));
memset(vis, 0, sizeof(vis));
dis[k] = 0;
q.push(make_pair(dis[k], k));
while (!q.empty()) {
int u = q.top().second; q.pop();
if (vis[u]) continue;
vis[u] = 1;
for (int i = head[u]; i; i = edge[i].next) {
int v = edge[i].to;
if (dis[v] > dis[u] + edge[i].dis) {
dis[v] = dis[u] + edge[i].dis;
if (!vis[v]) q.push(make_pair(dis[v], v));
}
}
}
}
int main() {
n = read(), m = read(), k = read(), s = read(), t = read();
for (int i = 1; i <= m; ++i) {
int a, b, c;
a = read(), b = read(), c = read();
for (int j = 0; j <= k; ++j) {
add(point(a, j), point(b, j), c);
add(point(b, j), point(a, j), c);
}
for (int j = 0; j <= k; ++j) {
add(point(a, j), point(b, j + 1), 0);
add(point(b, j), point(a, j + 1), 0);
}
}
dijkstra(s);
for (int i = 0; i <= k; ++i) {
ans = min(ans, dis[point(t, i)]);
}
printf("%d\n", ans);
// system("PAUSE");
}