Cod sursa(job #2706908)

Utilizator vlad_butnaruVlad Butnaru vlad_butnaru Data 16 februarie 2021 08:33:26
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <bits/stdc++.h>
using namespace std;
ifstream in ("stramosi.in");
ofstream out ("stramosi.out");
vector <int> v[250001];
int tata[250001];
int pd[25][250001];
int main ()
{
    int n, q;
    in >> n >> q;
    for (int i = 1; i<= n; ++i)
    {
        int a;
        in >> a;
        v[a].push_back(i);
        tata[i] = a;
    }
    for (int i = 1;i<=n;++i)
        pd[0][i] = tata[i];
    for(int j = 1;j<=25;++j)
        for (int i = 1; i<=n; ++i)
            pd[j][i] = pd[j-1][pd[j-1][i]];
    for (int i = 1; i<=q; ++i)
    {
        int a, b;
        in >> a >> b;
        int k = 25;
        while (k>=0 && b)
        {
            if (b & (1<<k))
            {
                b -=  (1<<k);
                a = pd[k][a];
            }
            k--;
        }
        out << a << '\n';
    }
    return 0;

}