Cf685b find the center of gravity of each subtree of a rooted tree

The main idea of the topic :

The title is very clear .

Topic ideas :

According to the nature of the center of gravity of the tree : Two subtrees merge , The center of gravity of the new tree must be on the connecting line of the centers of gravity of the two subtrees .

So when merging subtrees , Let the center of gravity of the subtrees jump to their ancestors . Until the end of jumping to the center of gravity .
PS: Determine the center of gravity : 2 * The largest subtree sz <= sz[u] that will do .

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vll vector<ll>
#define fi first
#define se second
const int maxn = 3e5 + 5;
const int mod = 1e9 + 7;
int sz[maxn] , res[maxn] , f[maxn] , mx[maxn];
vi e[maxn];
void dfs (int u , int fa){
    
    sz[u] = 1; res[u] = u;f[u] = fa;
    for (auto & v : e[u]){
    
        if (v == fa) continue;
        dfs(v , u);
        sz[u] += sz[v];
        mx[u] = max(mx[u] , sz[v]);
    }
    for (auto & v : e[u]){
    
        if (v == fa) continue;
        int now = res[v];
        //  Burst jump 
        bool ok = false;
        do{
    
            int tmx = max(mx[now] , sz[u] - sz[now]);
            if (2 * tmx <= sz[u]){
    
                res[u] = now;
                ok = true;
                break;
            }
            now = f[now];
            if (ok) break;
        }while (now != u);
    }
    return ;
}
int main()
{
    
    ios::sync_with_stdio(false);
    int n , m; cin >> n >> m;
    for (int i = 2 ; i <= n ; i++) {
    
        int x; cin >> x;
        e[x].pb(i);
    }
    dfs(1 , 0);
    for (int i = 1 ; i <= m ; i++){
    
        int x; cin >> x;
        cout << res[x] << endl;
    }
    return 0;
}