Educational Codeforces Round 136 (Rated for Div. 2) D. Reset K Edges

传送门：Problem - D - Codeforces

题意：给一棵树，k次修改，求最小深度，修改方式——选一个点删除其父节点把父节点改为根节点。

reference：Submission #173966040 - Codeforces

分析：我补到这个题的时候已经被队友群友等剧透完了。二分+贪心
首先在0到n之间二分答案，然后拿着这个答案去贪心的操作，统计操作数。

具体地，从下往上看（遍历，如果遍历到的节点子树的高度为我们当前的答案-1，则分两种情况：

1. 如果当前节点是根节点，那么是合法的（高度刚好是答案），无需操作
2. 如果当前节点不是根节点，那么需要进行操作，把当前节点删除，子树接到根节点

代码如下：

#include<bits/stdc++.h>
using namespace std;
 
#define ll long long
const int N = 2e5+5;
int fa[N];
 
int main(){
    // init();
    int T;
    cin >> T;
    while(T--){
        ll n, k;
        cin >> n >> k;
        for(int i = 1; i < n ;i ++){
            cin >> fa[i];
            fa[i]--;  // 非根节点从1到n，所以所有节点-1
        }
        int l = 0, r = n;
        while(l < r-1){
            // cout << l << " " << r << endl;
            int mid = (l + r)>>1;
            int h[n+1] = {0};
            int step = 0;
            for(int i = n-1; i > 0;i --){
                if(h[i] == mid - 1 && fa[i]) step ++;
                else h[fa[i]] = max(h[fa[i]], h[i] + 1);
            }
            // cout << step << endl;
            if(step <= k) r = mid;
            else l = mid;
        }
        // cout << l << " " << r << endl;
        cout << r << endl;
    }
}