Cod sursa(job #3253957)

Utilizator iuliageambazuGeambazu Iulia iuliageambazu Data 5 noiembrie 2024 16:25:49
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.85 kb
#pragma GCC optimize("O3", "Ofast", "unroll-loops")
#include <bits/stdc++.h>
#define int long long
using namespace std;

const int NMAX=250000,LOGMAX=20;
int n,m;
vector<int> logg(NMAX+5);
vector<vector<int>>anc(NMAX+5, vector<int>(LOGMAX+1));

signed main()
{
    freopen("stramosi.in","r",stdin);
    freopen("stramosi.out","w",stdout);
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;++i)
        cin>>anc[i][0];
    for(int i=2;i<=n;++i)
        logg[i]=logg[i/2]+1;
    for(int i=1;i<=n;++i)
        for(int j=1;j<=LOGMAX;++j)
        anc[i][j]=anc[anc[i][j-1]][j-1];
    for(int i=1;i<=m;++i)
    {
        int x,y;
        cin>>x>>y;
        while(y)
        {
            int k=logg[y];
            x=anc[x][k];
            y-=(1<<k);
        }
        cout<<x<<'\n';
    }
    return 0;
}