Cod sursa(job #2778781)

Utilizator tryharderulbrebenel mihnea stefan tryharderul Data 2 octombrie 2021 10:43:55
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <bits/stdc++.h>
#define NMAX 250003
#define log 19

using namespace std;

int n, m;
vector<vector<int> > g(NMAX);
int d[NMAX][log];

void dfs(int nod) {
    for(int i = 1; i < log; i++)
        d[nod][i] = d[d[nod][i-1]][i-1];
    for(int child : g[nod]) {
        dfs(child);
    }
}

int k_ancestor(int x, int y) {
    for(int i = log-1;i >= 0; i--)
        if((1<<i) & y)
            x = d[x][i];
    return x;

}

int main()
{
    freopen("stramosi.in","r",stdin);
    freopen("stramosi.out","w",stdout);
    scanf("%d %d",&n,&m);
    for(int i = 1; i <= n; i++) {
        int t;
        scanf("%d",&t);
        g[t].push_back(i);
        d[i][0] = t;
    }
    dfs(0);
    while(m--) {
        int x, y;
        scanf("%d %d",&x,&y);
        printf("%d\n",k_ancestor(x,y));
    }
    return 0;
}