洛谷:通往二分的WA之路
一开始写图论二分题,都是 好狠,怎么可能想得到 但写多了就会发现其实都有点规律:
- 求解的都是极值,比如至多/至少
- 有多个限制条件,比如本题就是 限制血量(边权)、限制点权(路径最大的最小) ;通信电路题限制的是,将不超过K条边变免费。在这些条件的限制下求极值
归根结底还是二分不熟练,题见太少了
以下代码(注释都是好东西):
#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define cout_double(a) cout << setiosflags(ios::fixed) << setprecision(a)
#define sca scanf
#define pri printf
#define ul u << 1
#define ur u << 1 | 1
//#pragma GCC optimize(2)
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 10010, M = 100010;
int INF = 0x3f3f3f3f, mod = 100003;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S;
int h[N], e[M], ne[M], w[M], idx;
ll dist[N];
bool st[N];
int d[N], sd[N];
//比较经典的二分答案图论题(想不到啊)
void add(int a, int b, int x) {
e[idx] = b, ne[idx] = h[a], w[idx] = x, h[a] = idx++;
}
bool check_dj(int sd_mid) {
mem(st, 0);
mem(dist, 0x3f);//初始化正无穷
dist[1] = 0;
priority_queue<PII, vector<PII>, greater<PII>> q;//堆优化
q.push({ 0,1 });
while (q.size())
{
PII t = q.top();
q.pop();
int ver = t.second;
if (st[ver])continue;
st[ver] = true;
for (int i = h[ver]; ~i; i = ne[i]) {
int j = e[i];
if (d[j] <= sd_mid && dist[j] > dist[ver] + w[i]) {
// sd_mid相当于天花板的存在,限制了最大点权的大小
// 可以更新点 == 可以走过去 == 该点权不能超过天花板
dist[j] = dist[ver] + w[i];
q.push({ dist[j],j });
}
}
}
//由此维护一条最短路,为什么要维护最短路?因为边权走越大意味着扣越多血
//应该尽可能保持血量健康
if (dist[n] > S)return false;
//二分的判断条件是:
//
//最短路径上扣的血与初始血量值作比较,显然大于代表没走到终点就 die 了
//这种状况下 mid 的肯定是小了,因为限制了可以更新的点
//mid 左边(本身 die 了显然也不能取)也更小,所以应该右移 l = mid + 1
return true;
//否则一旦满足了就停止在此,因为答案要求满足条件的 最小值
//所以左移 r = mid
}
int main() {
cinios;
cin >> n >> m >> S;
mem(h, -1);
for (int i = 1; i <= n; i++)cin >> d[i], sd[i] = d[i];
sort(sd + 1, sd + 1 + n);
//处理成有序的,对有序数据才能二分,答案也必定出自点权集合中
for (int i = 1; i <= m; i++) {
int a, b, x;
cin >> a >> b >> x;
add(a, b, x);
add(b, a, x);
}
if (!check_dj(INF))cout << "AFK"; //因为有可能费用选最大值也被干碎(怎么走血量都会减为0)
//所以先手预判一波,mid的含义就是 “ 路径上最大的点权的最小值(不止一条路径)”
else {
int l = 1, r = n;
while (l < r)
{
int mid = l + r >> 1;
if (check_dj(sd[mid]))r = mid;//二分第一个难点是联系到二分,证明其正确
//第二个难点就是二分情况的判断
else l = mid + 1;
}
cout << sd[l];
}
return 0;
}