Cod sursa(job #3206790)

Utilizator ililogIlinca ililog Data 24 februarie 2024 09:48:03
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
using namespace std;
#include<iostream>
#include<fstream>

#define NMAX 250005

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

int n, q;
int dp[40][NMAX];
int a, p;

int main() {
    
    fin >> n >> q;
    for (int i = 1; i<=n; i++) {
        fin >> dp[0][i];
    }
    
    for (int i = 1; (1<<i)<=n; i++) {
        for (int j = 1; j<=n; j++) {
            if (dp[i-1][j]) {
                dp[i][j] = dp[i-1][dp[i-1][j]];
            }
        }
    }
    
    for (int i = 1; i<=q; i++) {
        fin >> a >> p;
        
        int str = a;
        int d = 0;
        
        while (p) {
            if (p%2 == 1) {
                str = dp[d][str];
            }
            d++;
            p = p/2;
        }
        fout << str << '\n';
    }
    
    return 0;
}