Cod sursa(job #3261495)

Utilizator AndreiNicolaescuEric Paturan AndreiNicolaescu Data 6 decembrie 2024 13:57:17
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin("stramosi.in");
ofstream cout("stramosi.out");
int n, m, x, y;
vector<vector<int>> anc;
void binarylifting()
{
    for(int l=1; l<20; l++)
        for(int i=1; i<=n; i++)
            anc[i][l] = anc[anc[i][l-1]][l-1];
}
int ans(int node, int nanc)
{
    if(nanc >= (1<<20))
        return 0;
    for(int l=0; l<20; l++)
        if(nanc & (1<<l))
            node = anc[node][l];
    return node;
}
int main()
{
    cin >> n >> m;
    anc.assign(n+1, vector<int>(20, 0));
    for(int i=1; i<=n; i++)
        cin >> anc[i][0];
    binarylifting();
    while(m--)
    {
        cin >> x >> y;
        cout << ans(x, y) << '\n';
    }
    return 0;
}