Cod sursa(job #2498770)

Utilizator PatrickCplusplusPatrick Kristian Ondreovici PatrickCplusplus Data 24 noiembrie 2019 13:05:13
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <bits/stdc++.h>
#define NMAX 250000
using namespace std;

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

int n, m, dp[NMAX + 5][20];

int Query(int x, int y)
{
    int p = 0, sol = 0;
    while (y)
    {
        if (y % 2 == 1)
        {
            sol = dp[x][p];
            x = dp[x][p];
        }
        y = y / 2;
        ++p;
    }
    return sol;
}

int main()
{
    fin >> n >> m;
    for (int i = 1; i <= n; ++i)
    {
        int val;
        fin >> val;
        dp[i][0] = val;
    }
    for (int j = 1; j <= 18; ++j)
    {
        for (int i = 1; i <= n; ++i)
        {
            dp[i][j] = dp[dp[i][j - 1]][j - 1];
        }
    }
    while (m--)
    {
        int x, y;
        fin >> x >> y;
        fout << Query(x, y) << "\n";
    }
    fin.close();
    fout.close();
    return 0;
}