在树上选 \(m\) 条不重合的路径(可以有交点),使得这些路径长度的最小值最大。
看到最小值最大,很自然想到二分模型:枚举最小值 \(L\),看大于等于 \(L\) 的路径能不能有 \(m\) 条。
如何在树上选出 \(m\) 条路径最优成为我们要思考的问题,考虑树上贪心。
普遍的思路是从叶子节点按子树合并,我们要让该子树对父亲的贡献最大,一定是在该子树已经无法拼出路径(贪心)后剩下的路径中取最大值(由于父亲唯一,所以只能留下最大值)。
设 \(f(u)\) 为 \(u\) 子树内的节点到 \(u\) 节点的路径中无法拼出赛道的最长路径,我们已知所有的 \(f(v)+w(u,v)\),对于 \(f(v)+w(u,v)\ge L\) 的,直接贡献+1,对于剩下的,我们可以先让小的路径得到匹配,无法匹配的就更新 \(f(u)\)。这个过程可以用 multiset 维护。
复杂度为 \(O(n\log n\log\sum l)\)。
总结:二分模型,树形贪心的基本思路,合适的数据结构维护。
#include <bits/stdc++.h>
typedef long long ll;
int read() {
int x = 0, f = 1;
char c = getchar();
while(!isdigit(c)) {
if(c == '-') f = -1;
c = getchar();
}
while(isdigit(c)) {
x = (x << 3) + (x << 1) + (c - '0');
c = getchar();
}
return x * f;
}
int n, m, cnt, L, ret, ans;
int h[50010], f[50010];
struct node {
int to, nxt, w;
} e[100010];
void add(int u, int v, int w) {
e[++cnt].to = v;
e[cnt].nxt = h[u];
e[cnt].w = w;
h[u] = cnt;
}
void dfs(int u, int fa) {
std::multiset<int> s;
for(int i = h[u]; i; i = e[i].nxt) {
int v = e[i].to;
if(v == fa) continue;
dfs(v, u);
if(f[v] + e[i].w >= L) ret++;
else s.insert(f[v] + e[i].w);
}
while(!s.empty()) {
std::multiset<int>::iterator it = s.begin();
s.erase(it);
std::multiset<int>::iterator it2 = s.lower_bound(L - *it);
if(it2 == s.end()) {
f[u] = std::max(f[u], *it);
}
else {
s.erase(it2);
ret++;
}
}
}
bool check(int x) {
L = x;
ret = 0;
memset(f, 0, sizeof(f));
dfs(1, 0);
return ret >= m;
}
void Solve() {
n = read(), m = read();
int r = 0;
for(int i = 1; i < n; i++) {
int u = read(), v = read(), w = read();
add(u, v, w), add(v, u, w);
r += w;
}
int l = 0;
r = r / m;
while(l <= r) {
int mid = (l + r) >> 1;
if(check(mid)) l = mid + 1, ans = mid;
else r = mid - 1;
}
std::cout << ans << "\n";
}
int main() {
Solve();
return 0;
}
标签:赛道,std,int,路径,ret,NOIP2018,read,P5021
From: https://www.cnblogs.com/FireRaku/p/18091679