Cod sursa(job #3132981)

Utilizator RolandPetreanPetrean Roland RolandPetrean Data 24 mai 2023 17:00:33
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.63 kb
// https://infoarena.ro/problema/stramosi
#include <bits/stdc++.h>
using namespace std;
#define endl '\n'

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

const int MAXN=250005;
int p[MAXN], anc[18][MAXN];

int main() {
  int n, m;
  fin>>n>>m;

  for (int i=1; i<=n; ++i) {
    int x;
    fin>>x;
    p[i] = x;
  }

  for (int i=1; i<=n; ++i) anc[0][i] = p[i];
  for (int h=1; h<18; ++h) {
    for (int i=1; i<=n; ++i) anc[h][i] = anc[h-1][anc[h-1][i]];
  }

  while (m--) {
    int q, i;
    fin>>q>>i;

    int curr=q;
    for (int j=0; j<18; ++j) {
      if (i&(1<<j)) curr = anc[j][curr];
    }

    fout<<curr<<endl;
  }
}