Cod sursa(job #2837407)

Utilizator robertanechita1Roberta Nechita robertanechita1 Data 22 ianuarie 2022 10:28:45
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <bits/stdc++.h>

using namespace std;

/// d[i][j] - stramosul lui j aflat la distanta 2^i de j

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

int n, q, d[20][250003];

int main()
{
    int nod, k, sol;
    fin >> n >> q;
    for(int i = 1; i <= n; i++)
        fin >> d[0][i];

    for(int i = 1; i <= 18; i++)
        for(int j = 1; j <= n; j++)
            d[i][j] = d[i-1][ d[i-1][j] ];

    for(int i = 1; i <= q; i++)
    {
        fin >> nod >> k;
        sol = 0;
        while(k > 0)
        {
            if(k % 2 == 1)
                nod = d[sol][nod];
            sol++;
            k /= 2;
        }
        fout << nod << "\n";
    }
    return 0;
}