Cod sursa(job #2564622)

Utilizator ursu0406Ursu Ianis-Vlad ursu0406 Data 2 martie 2020 08:23:22
Problema Stramosi Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <bits/stdc++.h>
using namespace std;

const int MAX_N = 25e4 + 1;
const int LOG_MAX_N = 18;

int binary_lifting[LOG_MAX_N][MAX_N], N, M;

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


int main()
{
    fin >> N >> M;

    for(int i = 1; i <= N; ++i)
        fin >> binary_lifting[0][i];

    for(int k = 1; k <= LOG_MAX_N; ++k)
        for(int i = 1; i <= N; ++i)
            binary_lifting[k][i] = binary_lifting[k - 1][binary_lifting[k - 1][i]];

    for(int i = 1; i <= M; ++i)
    {
        int q, p;
        fin >> q >> p;

        for(int j = 0; q != 0 && j <= LOG_MAX_N; ++j)
        {
            if((1 << j) & p){
                q = binary_lifting[j][q];
            }
        }

        fout << q << '\n';
    }

}