Cod sursa(job #2283176)

Utilizator TooHappyMarchitan Teodor TooHappy Data 15 noiembrie 2018 01:46:26
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <bits/stdc++.h>
 
using namespace std;
 
ifstream in("stramosi.in");
ofstream out("stramosi.out");
 
const int Nmax = 25e4;
 
int stramos[20][Nmax + 10];

int getStramos(int q, int p) {
    while(p) {
        int k = log2(p);
        q = stramos[k][q];
        p -= (1 << k);
    }

    return q;
}
 
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[0][i];
    }
 
    for(int i = 1; (1 << i) <= n; ++i) {
        for(int j = 1; j <= n; ++j) {
            stramos[i][j] = stramos[i - 1][stramos[i - 1][j]];
        }
    }
 
    while(m--) {
        int q, p; in >> q >> p;

        out << getStramos(q, p) << '\n';
    }
 
    in.close(); out.close();
 
    return 0;
}