Cod sursa(job #2283120)

Utilizator TooHappyMarchitan Teodor TooHappy Data 14 noiembrie 2018 23:58:44
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.76 kb
#include <iostream>
#include <fstream>
 
using namespace std;
 
ifstream in("stramosi.in");
ofstream out("stramosi.out");

const int Nmax = 250010;

int stramos[Nmax][18];

int main() {
    ios::sync_with_stdio(false); in.tie(0); out.tie(0);

    int n, m; in >> n >> m;

    for(int i = 1; i <= n; ++i) {
        in >> stramos[i][0];
    }

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

    for(int i = 1; i <= m; ++i) {
        int q, p; in >> q >> p;

        for(int i = 0; i < 18 && q; ++i) {
            if((p >> i) % 2 == 1) {
                q = stramos[q][i];
            }
        }
        out << q << "\n";
    }

    in.close(); out.close();
 
    return 0;
}