Cod sursa(job #3349058)

Utilizator robertcosacCosac Robert-Mihai robertcosac Data 25 martie 2026 11:06:09
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.68 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("stramosi.in");
ofstream g("stramosi.out");
int str[22][300009];
int query (int nod, int niv)
{
    int ans=nod;
    for (int bit=20; bit>=0; bit--)
    {
        int p=(1<<bit);
        if (niv&p)
            ans=str[bit][ans];
    }
    return ans;
}
signed main ()
{
    int n, q;
    f >> n >> q;
    for (int i=1; i<=n; i++)
        f >> str[0][i];
    for (int niv=1; (1<<niv)<=n; niv++)
    {
        for (int i=1; i<=n; i++)
            str[niv][i]=str[niv-1][str[niv-1][i]];
    }
    while (q--)
    {
        int nod, niv;
        f >> nod >> niv;
        g << query (nod, niv) << '\n';
    }

}