Cod sursa(job #3343515)

Utilizator Bogdan_RuscanuRuscanu Stefan Bogdan Bogdan_Ruscanu Data 27 februarie 2026 17:23:02
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.95 kb
#include <fstream>
const int NMAX=250005;
const int LOG=19;

using namespace std;

ifstream cin("stramosi.in");
ofstream cout("stramosi.out");

int b[LOG][NMAX]; ///b[i][j] = al 2^i-lea stramos al lui j

void build(int n)
{
    for(int i=1;(1<<i)<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            b[i][j]=b[i-1][b[i-1][j]];
        }
    }
}

int query(int nod,int k)
{
    int i;
    for(i=1;(1<<i)<=k;i++) {}
    i--;
    for(;i>=0;i--)
    {
        if(k>=(1<<i))
        {
            nod=b[i][nod];
            k-=(1<<i);
        }
    }
    return nod;
}

int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int nr; ///stramosul lui i este nr
        cin>>nr;
        b[0][i]=nr;
    }
    build(n);
    for(int i=1;i<=m;i++)
    {
        int nod,k;
        cin>>nod>>k; ///care este al k-lea stramos pentru nod
        cout<<query(nod,k)<<endl;
    }
    return 0;
}