ABC267 F - Exactly K Steps

感觉根本不会树论，等篮球被过后再学！

F - Exactly K Steps (atcoder.jp)

题意：

思路：

对于树上一个点跳到定长距离的点，最常见的做法就是树上倍增

而树上的结点有一个很重要的性质：在这个连通图中和它距离最远的结点一定是直径的两端点之一

因此我们一定向最长的方向跳，即往直径的端点跳

那么就需要知道哪两个点是直径的端点，这个求一下树的直径即可

求完直径之后，由直径的两个端点作为根建树，然后再倍增地跳定长距离k即可

Code：

#include <bits/stdc++.h>
using namespace std;
#define int long long 
const int mxn=2e5+10;
const int mxe=2e5+10;
const int mod=1e9+7;
struct ty{
    int to,next;
}edge[mxe<<2];
struct Query{
    int u,k;
}q[mxn];

int n,Q,u,v,tot=0;
int head[mxn],dep[mxn],f[mxn][33],ans[mxn];
void add(int u,int v){
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void G_init(){
    tot=0;
    for(int i=0;i<=n;i++){
        head[i]=-1;
    }
}
void dfs(int u,int fa){
    dep[u]=dep[fa]+1;
    f[u][0]=fa;
    for(int j=1;j<=30;j++) f[u][j]=f[f[u][j-1]][j-1];
    for(int i=head[u];~i;i=edge[i].next){
        if(edge[i].to==fa) continue;
        dfs(edge[i].to,u);
    }
}
int find(int u,int k){
    for(int j=30;j>=0;j--){
        if((k>>j)&1) u=f[u][j];
    }
    return u?u:-1;
}
void work(int u){
    dfs(u,0);
    for(int i=1;i<=Q;i++){
        if(ans[i]==-1) ans[i]=find(q[i].u,q[i].k);
    }
}
void solve(){
    memset(ans,-1,sizeof(ans));
    cin>>n;
    G_init();
    for(int i=1;i<=n-1;i++){
        cin>>u>>v;
        add(u,v);
        add(v,u);
    }
    int d1=1,d2=1;
    dfs(1,0);
    for(int i=1;i<=n;i++){
        if(dep[d1]<dep[i]) d1=i;
    }
    dfs(d1,0);
    for(int i=1;i<=n;i++){
        if(dep[d2]<dep[i]) d2=i;
    }
    cin>>Q;
    for(int i=1;i<=Q;i++){
        cin>>q[i].u>>q[i].k;
    }
    work(d1);
    work(d2);
    for(int i=1;i<=Q;i++) cout<<ans[i]<<" \n";
}
signed main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int __=1;//cin>>__;
    while(__--)solve();return 0;
}