Cod sursa(job #3293135)

Utilizator rrfeierFeier Raul rrfeier Data 10 aprilie 2025 13:48:35
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.64 kb
#include <bits/stdc++.h>
#include <fstream>

using namespace std;

int parents[30][250005];

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);

  ifstream cin{"stramosi.in"};
  ofstream cout{"stramosi.out"};

  int N, Q;
  cin >> N >> Q;

  for (int i = 1; i <= N; i++) {
    cin >> parents[0][i];
  }

  for (int i = 1; i <= 29; i++) {
    for (int j = 1; j <= N; j++) {
      parents[i][j] = parents[i - 1][parents[i - 1][j]];
    }
  }

  while (Q--) {
    int a, b;
    cin >> a >> b;

    for (int i = 0; i <= 29; i++) {
      if (b & (1 << i)) {
        a = parents[i][a];
      }
    }

    cout << a << '\n';
  }

  return 0;
}