Cod sursa(job #1923111)

Utilizator micutuAndrei Vasile micutu Data 10 martie 2017 20:47:21
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <iostream>
#include <fstream>
#define NMAX 250005
#define KMAX 20
using namespace std;

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

int n, m, stramos[KMAX][NMAX];

int main ()
{
    fin >> n >> m;
    for (int i = 1; i <= n; ++i) {
        fin >> stramos[0][i];
        //cout << log[i] << " ";
    }
    for (int k = 1; (1 << k) <= n; ++k) {
        for (int i = 1; i <= n; ++i) {
            stramos[k][i] = stramos[k-1][stramos[k-1][i]];
        }
    }
    int q, p;
    for (int i = 1; i <= m; ++i) {
        fin >> q >> p;
        for (int j = 0; (1 << j) <= p; ++j) {
            if ((1 << j) & p) {
                q = stramos[j][q];
            }
        }
        fout << q << '\n';
    }
    
    fin.close();
    fout.close();
    return 0;
}