Cod sursa(job #2507829)

Utilizator AlexandruGabrielAliciuc Alexandru AlexandruGabriel Data 10 decembrie 2019 21:49:43
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <bits/stdc++.h>
#define N 250005

using namespace std;

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

int n, m;
int dp[N][20], lg2[N];

void calcLog()
{
    for (int i = 2; i < N; i++)
        lg2[i] = lg2[i / 2] + 1;
}

void calcDp()
{
    for (int k = 1; k <= lg2[n]; k++)
    {
        for (int nod = 1; nod <= n; nod++)
                dp[nod][k] = dp[dp[nod][k - 1]][k - 1];
    }

}

int Query(int nod, int p)
{
    int nodAns = nod;

    for (int i = 0; i <= lg2[p]; i++)
    {
        if (p & (1 << i))
            nodAns = dp[nodAns][i];
    }

    return nodAns;
}

int main()
{
    fin >> n >> m;
    for (int i = 1; i <= n; i++)
        fin >> dp[i][0];

    calcLog();
    calcDp();

    for (int i = 1; i <= m; i++)
    {
        int q, p;
        fin >> q >> p;
        fout << Query(q, p) << "\n";
    }
    return 0;
}