【ybt 入门到进阶模拟赛 Day1 T4】通信线路

通信线路

题目链接：ybt 入门到进阶模拟赛 Day1 T4

题目大意

给你一个无向图，边有边权，然后你要找一条路径从 1 到 n，让它第 k+1 大的边最小。
如果不存在路径，就输出 -1。

思路

直接搞似乎很难，我们考虑二分要输出的答案。

假设你二分的是 $mid$。
那你就相当于要找一条路径，使得这条路径上长度大于 $mid$ 的边不超过 $k$ 个。
那我们就把原来图中边权大于 $mid$ 看做边权是 $1$，小于等于 $mid$ 看做边权是 $0$，然后跑 SPFA，然后看 $di{s}_n$ 时候小于等于 $k$，如果是就代表 $mid$ 可以，否则就不可以。

然后最后输出二分出的值就可以了。

代码

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

struct node {
	int x, to, nxt;
}e[20001];
int n, p, k, x, y, z;
int l, r, ans, mid;
int le[10001], KK;
int dis[10001];
bool in[10001];
queue <int> q;

void add(int x, int y, int z) {
	e[++KK] = (node){z, y, le[x]}; le[x] = KK;
	e[++KK] = (node){z, x, le[y]}; le[y] = KK;
}

bool ck(int mid) {
	memset(dis, 0x7f, sizeof(dis));
	memset(in, 0, sizeof(in));
	while (!q.empty()) q.pop();
	
	dis[1] = 0;
	q.push(1);
	in[1] = 1;
	
	while (!q.empty()) {
		int now = q.front();
		q.pop();
		
		for (int i = le[now]; i; i = e[i].nxt)
			if (dis[e[i].to] > dis[now] + (e[i].x > mid)) {//这条边的长度大于你二分的，那就说明它一定要被免费（也就是一定要位于前 k 个位置）
				dis[e[i].to] = dis[now] + (e[i].x > mid);
				if (!in[e[i].to]) {
					in[e[i].to] = 1;
					q.push(e[i].to);
				}
			}
		
		in[now] = 0; 
	}
	
	return dis[n] <= k;//如果要位于前 k 个位置的小于等于 k 个，那就是可以的，否则你就摆不下那么多，就不可以
}

int main() {
//	freopen("phone.in", "r", stdin);
//	freopen("phone.out", "w", stdout);
	
	scanf("%d %d %d", &n, &p, &k);
	for (int i = 1; i <= p; i++) {
		scanf("%d %d %d", &x, &y, &z);
		add(x, y, z);
		r = max(r, z);
	}
	
	ans = -1;
	l = 0;
	while (l <= r) {//二分要输出的长度
		mid = (l + r) >> 1;
		if (ck(mid)) {
			ans = mid;
			r = mid - 1;
		}
		else l = mid + 1;
	}
	
	printf("%d", ans);
	
	return 0;
}