知识点:二分,最短路
这个题还可以用动态规划+最短路的做法,但是动态规划没学,就用二分+最短路的方法,首先这个题可以用二分,显然具有单调性,然后是判定函数,这个非常巧妙,我们求起点到终点的最短路,小于等于二分值的边长为0,大于二分值的边的边长为1,最后得到最短路的长度小于等于k,那么返回true,否则返回false,这个是01最短路,用双端队列BFS求解效率最高,所以这个题的算法就是二分+双端队列BFS,感觉还是很巧妙的,第一次看没有想出来就看了李煜东的讲解,希望自己以后做题的时候也能看出来
刚看了题解,里面一个提炼的模型是,求起点到终点的一条路径,使得路径上第K+1长的边尽量短,能想到这个模型其实离想到二分就不远了,但是这个模型本身就包含了贪心的思想在里面,不是说把题意概括一下就想到的
从这个题可以感受到01最短路都不会直接给你一个图边长是0或1,一般都是需要间接的转化一下,甚至可以完成一下计数的功能
#include <bits/stdc++.h>
using namespace std;
const int N = 1005, M = 2e4 + 5;
struct node {
int x, d;
node() {}
node(int a, int b): x(a), d(b) {}
};
int tot, ver[M], edge[M], nxt[M], head[N];
int n, m, k, vis[N];
void add(int x, int y, int z) {
ver[++tot] = y; edge[tot] = z;
nxt[tot] = head[x]; head[x] = tot;
}
bool bfs(int max_d) {
memset(vis, 0, sizeof(vis));
deque<node> q;
q.push_front(node(1, 0));
while (!q.empty()) {
node now = q.front(); q.pop_front();
if (now.x == n) return now.d <= k;
if (vis[now.x]) continue;
vis[now.x] = 1;
for (int i = head[now.x]; i; i = nxt[i]) {
if (edge[i] <= max_d) q.push_front(node(ver[i], now.d));
else q.push_back(node(ver[i], now.d + 1));
}
}
return false;
}
int main() {
cin >> n >> m >> k;
int l = 0, r = -1;
for (int i = 1; i <= m; i++) {
int x, y, z;
scanf("%d%d%d", &x, &y, &z);
r = max(r, z + 1);
add(x, y, z);
add(y, x, z);
}
int num = r;
while (l < r) {
int mid = (l + r) >> 1;
if (bfs(mid)) r = mid;
else l = mid + 1;
}
cout << (l == num ? -1 : l);
return 0;
}