Cod sursa(job #2321620)

Utilizator OldpugAlex Ionescu Oldpug Data 16 ianuarie 2019 13:18:53
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.53 kb
#include <fstream>

int mat[25][250001];

int main() {
  std::ifstream is{"stramosi.in"};

  int n, m;
  is >> n >> m;

  for (auto i = 1; i <= n; ++i)
    is >> mat[0][i];

  for (auto k = 1; (1 << k) <= n; ++k)
    for (auto j = 1; j <= n; ++j)
      mat[k][j] = mat[k - 1][mat[k - 1][j]];

  std::ofstream os{"stramosi.out"};

  while (m--) {
    int q, p, exp{};
    is >> q >> p;

    while (p) {
      if (p & 1)
        q = mat[exp][q];

      ++exp;
      p >>= 1;
    }

    os << q << '\n';
  }
}