Codeforces Round 936 (Div. 2)------＞C. Tree Cutting

一，思路：

1.切k次就会产生 k+1 个连通块。

2.通过dfs进行切除（如果你对 dfs() 求联通块大小很了解的化，这一点也能很快接受）。

3.二分答案，重些check() 函数。（具体看代码注释）。

二，代码：

#include <iostream>
#include<algorithm>
#include<cstring>
using namespace std;

const int N=2e5+10;

//数组模拟邻接表（vector也行）
int h[N],e[2*N],ne[2*N],idx;

//cnt[N]用于记录联通块大小
int cnt[N],n,k,res;

//建图模板
void add(int a,int b){
   e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

//dfs切边
int dfs(int v,int pa,int x){

//这里是求连通块大小的模板
    cnt[v]=1;
    for(int i=h[v];i!=-1;i=ne[i]){
        int t=e[i];
        if(t!=pa){
           cnt[v]+=dfs(t,v,x);
        }
    }

//这里加个判断，如果这个联通块的大小符合二分到的数量，则res++ (数量最小为x的联通块数量)
    if(cnt[v]>=x){
        res++;

//细品 return 0---->就相当于在这颗子树的上面来一刀，将他独立出去。
        return 0;
    }

    return cnt[v];
}

//二分比较函数
bool check(int x){
    res=0;
    dfs(1,0,x);

//当连通块数量大于等于k+1时返回true
    return res>k;
}

void Solved() {

    memset(h,-1,sizeof h);


    cin>>n>>k;

    for(int i=1;i<=n-1;i++){
        int a,b;
        cin>>a>>b;

//双向边
        add(a,b),add(b,a);
    }


//二分模板
    int l=1,r=n;

    while(l<r){
        int mid=(l+r+1)>>1;
        if(check(mid)) l=mid;
        else r=mid-1;
    }

    cout<<l<<endl;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);

    int t;
    cin>>t;

    while(t--) {
        Solved();
    }
    return 0;
}