题目描述:
思路:
为了在选择了最多k条边之后使得剩下的边中的最大值最小,我们在每次确定了一条路后,应当每一次都选择路上权值最大的边,将其纳入免费的k条边中。
所以,假设路上的边数>k,那么我们一定会选择这条路上的最大的k个数纳入免费的集合中。故剩下的边中最大的就是这条路上所有边中第k + 1大的数。
那么这道题就转化为了求每一条1~n的路上第k + 1大的数的最小值。
看到这种形式,我们就会想到用二分。而使用二分需要满足以下的性质:
对于数列中一个元素x,它可以将整个数列分割成左右两个有着不同性质的区间。
所以我们要想使用功二分,还需要在本题中找到一种性质。
这种性质是:
对于一个x,在图中一条1~n的路上的所有边中权值大于x的数量是否小于等于k。
这样就可以将整个区间[0 , 1000001]分成左右两个有着不同性质的区间了。
我们来证明一下:
假设本体答案为x。
对于右侧区间:
在x右侧任取一个数y,那么显然满足y > x。
又由于在x处满足1~n的路上所有权值大于x的边总数小于等于k,因此大于y的边的总数一定小于等于大于x的总数,故小于等于k。
即:
总数(y) <= 总数(x) <= k
对于左侧区间:
运用反证法。
假设左侧存在一个数z,使得z处满足本题中的性质。由于z < x,这与x的“最小的满足题意的数”的定义矛盾。
故假设不成立。
那么,怎么去判断一个是否满足这个性质呢?
我们可以这样来计算:
求从1到n最少经过的权值大于x的数量。
如果一条边,权值大于x,那么久把它的权值置为1;反之,则置为0;然后再用双端队列广搜来计算最短路即可。
代码:
#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
const int N = 1010, M = 2e4 + 10;
int h[N], e[M], ne[M], w[M], idx; //邻接表
int n, m, k;
deque<int> q; //双向队列
int dist[N];
bool st[N];
void add(int a, int b, int c) {
e[idx] = b;
ne[idx] = h[a];
w[idx] = c;
h[a] = idx ++ ;
}
bool check(int bound) { //计算最短路来判断是否满足性质
memset(dist, 0x3f, sizeof dist);
memset(st, 0, sizeof st);
dist[1] = 0;
q.push_back(1);
while (q.size()) {
int t = q.front();
q.pop_front();
if (st[t])
continue;
st[t] = true;
for (int i = h[t] ; ~i ; i = ne[i]) {
int j = e[i], v = w[i] > bound; //比较边权与x的大小
if (dist[j] > dist[t] + v) {
dist[j] = dist[t] + v;
if (!v)
q.push_front(j);
else
q.push_back(j);
}
}
}
return dist[n] <= k;
}
int main() {
scanf("%d%d%d", &n, &m, &k);
memset(h, -1, sizeof h);
while (m -- ) {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
add(a, b, c), add(b, a, c); //无向图
}
int l = 0, r = 1e6 + 1;
while (l < r) { //二分
int mid = l + r >> 1;
if (check(mid)) //判断是否满足性质
r = mid;
else
l = mid + 1;
}
if (r == 1e6 + 1)
puts("-1");
else
printf("%d\n", r);
return 0;
}
注意事项:
题目中二分的边界设置为0和10^6 + 1,有两点原因:
1.对于左端点0:
题目中是可以出现0这个答案的(例如一条路边数小于k),因此它必须出现在二分的区间中。
2.对于右端点1000001:
首先,题目说明了边权最大值为1000000,也就是说题目中是可以出现1000000这个答案的。
其次,题目明确说明了存在无解情况,即1与n不连通。此时如果直接二分,其答案也会为1000000,与上面这种情况难以分辨。所以我们的有边界应置为1000001,来作区分。