Cod sursa(job #3206800)

Utilizator Alexia_CiobanuAlexia Maria Ciobanu Alexia_Ciobanu Data 24 februarie 2024 10:02:18
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <fstream>
#include <iostream>

#define NMAX 250007

using namespace std;
ifstream fin("stramosi.in");
ofstream fout("stramosi.out");

int n, m;
int stramos[NMAX], dp[NMAX][40];

int k_stramos_pt_x(int x, int k)
{
    int anc=x;
    int d=0;
    while (k >> d)
    {
        if ((k >> d) % 2)
            anc = dp[anc][d];
        d++;
    }
    return anc;
}
int main()
{
    fin>>n>>m;
    for (int i = 1; i <= n; i++)
    {
        fin>>stramos[i];
        dp[i][0]=stramos[i];
    }
    for (int i = 1; (1 << i) <= n; i++)
        for (int j = 1; j <= n; j++)
            if(dp[j][i-1])
                dp[j][i] = dp[dp[j][i-1]][i-1];
    int q, p;
    for (int i = 1; i <= m; i++)
    {
        fin >> q >> p;
        fout << k_stramos_pt_x(q, p) << '\n';
    }
    return 0;
}