Cod sursa(job #2397625)

Utilizator DenisacheDenis Ehorovici Denisache Data 4 aprilie 2019 17:15:04
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <bits/stdc++.h>

using namespace std;

int D[250005][18];

int log(int n) {
    int res = 0;
    
    while (n) {
        ++res;
        n >>= 1;
    }
    
    return res;
}

int main() {
    ios::sync_with_stdio(false);
    
    ifstream fin("stramosi.in");
    ofstream fout("stramosi.out");
    
    int n, q;
    fin >> n >> q;
    
    for (int i = 1; i <= n; ++i) {
        fin >> D[i][0];
    }
    
    int logN = log(n);
    
    for (int j = 1; j < logN; ++j) {
        for (int i = 1; i <= n; ++i) {
            D[i][j] = D[D[i][j-1]][j-1];
        }
    }
    
    while (q--) {
        int x, p;
        fin >> x >> p;
        
        int pwr = 0;
        while (p && x) {
            if (p & 1) {
                x = D[x][pwr];
            }
            
            ++pwr;
            p >>= 1;
        }
        
        fout << x << "\n";
    }
    
    return 0;
}