CF1739D Reset K Edges rating1900

题目描述：

给出1~n的有根树，删除任意边（v，u）,添加一条（1，u）为一次操作，要求在至多k次操作内，树的高度（定义为树的最大深度）最小。

输入：

t组测试数据（1<t<1e4）

点数n，操作次数k（2<=n<=2e5,0<=k<=n-1）

pi表示节点i的父节点为i p2,p3,p4……pn（1<pi<i,2<=i<=n）

输出：

k次操作后最小的树的高度

Solution：

由于题目是求最大值中的最小值问题，故此处我们可以采用二分的解法，对于树的高度进行二分，判断在每次树高为mid的情况下，所需的操作次数res与k进行比较。

而求res的过程中，我们只需要遍历每个节点的最大深度，同时判断子节点的最大深度是否满足要求，如果满足则res++，否则更新父节点的最大深度。

C++Code：

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10, M = 4e5 + 10;
int n, k;
int res = 0;
int h[N], e[M], ne[M], idx = 0;
map<int, int>m;
void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void dfs(int u,int mid) {//遍历每个节点的最大深度，计算操作次数
    for (int i = h[u]; i != -1; i = ne[i]) {
        int j = e[i];
        dfs(j, mid);
        if (m[j] + 1 == mid)res++;//如果子节点符合要求便res++
        else m[u] = max(m[u], m[j] + 1);//否则更新父节点的最大深度
    }
}
bool check(int mid) {
    res = 0;
    m.clear();
    for (int i = h[1]; i != -1; i = ne[i]) {//遍历节点1的子节点
        int j = e[i];
        dfs(j, mid);
    }
    return res <= k;
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int t;
    cin >> t;
    while (t--) {
        idx = 0;
        memset(h, -1, sizeof h);
        cin >> n >> k;
        for (int i = 2; i <= n; i++) {
            int b;
            cin >> b;
            add(b, i);
        }
        int l = 1, r = n - 1;
        while (l < r) {//二分
            int mid = l + r >> 1;
            if (check(mid))r = mid;
            else l = mid + 1;
        }
        cout << r << endl;
    }
    return 0;
}