Cod sursa(job #3204966)

Utilizator v_mateiVatamanu Matei v_matei Data 18 februarie 2024 13:50:48
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <fstream>
#include <iostream>
#include <vector>

#define NMAX 250'007

std::ifstream cin("stramosi.in");
std::ofstream cout("stramosi.out");

int n, m;
int stramos[NMAX], dp[NMAX][40];

void citire() {
  cin >> n >> m;
  for (int i = 1; i <= n; i++) {
    cin >> stramos[i];
    dp[i][0] = stramos[i];
  }
}

int ancestor(int x, int k) {
    int anc = x;
    int d = 0;
    while (k >> d) {
      if ((k >> d) % 2) {
        anc = dp[anc][d];
      }
      d++;
    }
  return anc;
}

void queries() {
  int q, p;
  for (int i = 1; i <= m; i++) {
    cin >> q >> p;
    cout << ancestor(q, p) << '\n';
  }
}

int main() {
  citire();

  for (int i = 1; (1 << i) <= n; i++) {
    for (int j = 1; j <= n; j++) {
      if(dp[j][i-1])
        dp[j][i] = dp[dp[j][i-1]][i-1];
    }
  }
  queries();
  return 0;
};