cf980e The Number Games

挺简单的，从 \(n\) 到 \(1\) 探讨是否能被加入。要加入的话必定是加入一条链，倍增以下看看要加多少，能不能加就可以了。

#include <algorithm>
#include <iostream>
#include <cstdio>
using namespace std;
int n, k, hea[1000005], cnt, uu, vv, dep[1000005], fa[1000005][21];
bool vis[1000005];
struct Edge{
    int too, nxt;
}edge[2000005];
void add_edge(int fro, int too){
    edge[++cnt].nxt = hea[fro];
    edge[cnt].too = too;
    hea[fro] = cnt;
}
void dfs(int x, int f, int d){
    dep[x] = d;
    fa[x][0] = f;
    for(int i=1; i<=20; i++)
        fa[x][i] = fa[fa[x][i-1]][i-1];
    for(int i=hea[x]; i; i=edge[i].nxt){
        int t=edge[i].too;
        if(t!=f)
            dfs(t, x, d+1);
    }
}
int f(int x){
    int t=dep[x];
    for(int i=20; i>=0; i--)
        if(!vis[fa[x][i]])
            x = fa[x][i];
    return t-dep[x]+1;
}
int main(){
    cin>>n>>k;
    k = n - k - 1;
    for(int i=1; i<n; i++){
        scanf("%d %d", &uu, &vv);
        add_edge(uu, vv);
        add_edge(vv, uu);
    }
    dfs(n, 0, 1);
    vis[0] = vis[n] = true;
    for(int i=n-1; i && k; i--){
        if(vis[i])  continue;
        int tmp=f(i);
        if(tmp<=k){
            k -= tmp;
            for(int j=i; !vis[j]; j=fa[j][0])
                vis[j] = true;
        }
    }
    for(int i=1; i<=n; i++)
        if(!vis[i])
            printf("%d ", i);
    return 0;
}