Cod sursa(job #2491488)

Utilizator PatrickCplusplusPatrick Kristian Ondreovici PatrickCplusplus Data 12 noiembrie 2019 17:47:28
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <bits/stdc++.h>

using namespace std;

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

int n, m, x, y, dp[250005][20], logg[250005];

int Query(int a, int b)
{
    int putere = (1 << logg[b]);
    if (putere == b)
    {
        return dp[a][logg[b]];
    }
    return Query(dp[a][logg[b]], b - putere);
}

int main()
{
    fin >> n >> m;
    for (int i = 1; i <= n; ++i)
    {
        int nod;
        fin >> nod;
        dp[i][0] = nod;
        if (i == 1)
            continue;
        logg[i] = 1 + logg[i / 2];
    }
    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--)
    {
        fin >> x >> y;
        fout << Query(x, y) << "\n";
    }
    fin.close();
    fout.close();
    return 0;
}