Cod sursa(job #3206796)

Utilizator Alexia_CiobanuAlexia Maria Ciobanu Alexia_Ciobanu Data 24 februarie 2024 09:54:02
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.85 kb
#include <fstream>
#include <iostream>
#include <vector>

#define NMAX 250007

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

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

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

int k_stramos_pt_x(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++)
  {
    fin >> q >> p;
    fout << k_stramos_pt_x(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;
};