Cod sursa(job #2283112)

Utilizator TooHappyMarchitan Teodor TooHappy Data 14 noiembrie 2018 23:47:09
Problema Stramosi Scor 80
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");

vector< vector< int > > stramos;

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

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

    stramos.resize(n + 1, vector< int >(20));
    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; ++i) {
            if((p >> i) % 2 == 1) {
                q = stramos[q][i];
            }
        }
        out << q << "\n";
    }

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