Cod sursa(job #2976277)

Utilizator andreibrosPeta Andrei Mathias andreibros Data 8 februarie 2023 20:09:44
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream in("stramosi.in");
ofstream out("stramosi.out");
const int NMAX=250005;

int A[NMAX][30];
int tata[NMAX];
int stramos(int nod, int p)
{
        int i=0;
        while(p!=0)
        {
            if(p%2==1)
                nod=A[nod][i];
            i++;
            p=p/2;
        }
        return nod;
}

int main()
{
    int n,m;
    in>>n>>m;
    for(int i=1; i<=n; i++)
        in>>tata[i];

    for(int i=1; i<=n; i++)
        A[i][0]=tata[i];

    for(int i=1; i<=n; i++)
        for(int j=0; (1<<j)<=n; j++)
            if(A[i][j-1])
                A[i][j]=A[A[i][j-1]][j-1];
    int nod,p;
    for(int i=1; i<=m; i++)
    {
        in>>nod>>p;
        out<<stramos(nod,p)<<"\n";
    }


    return 0;
}