Cod sursa(job #2214431)

Utilizator AndreiDumitrescuAndrei Dumitrescu AndreiDumitrescu Data 19 iunie 2018 00:03:48
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.05 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream f("stramosi.in");
ofstream g("stramosi.out");

int n, m;
int dp[250010][30], tati[250010];

int creez_dinamica()
{
    for(int j = 1; j < 22; j++)
        for(int i = 0; i <= n; i++)
            dp[i][j] = dp[dp[i][j-1]][j-1];
}

int query(int x, int p)
{
    int putere = 1, contor = 0;
    while(putere <= p)
    {
        putere *= 2;
        contor++;
    }
    putere /= 2;
    contor--;
    if(putere == p)
        return dp[x][contor];
    else
        query(dp[x][contor], p - putere);

}

int main()
{
    int x, p;
    f >> n >> m;
    for(int i = 1; i <= n; i++)
    {
         f >> tati[i];
         dp[i][0] = tati[i];
    }
    creez_dinamica();
   /* for(int i = 1; i <= n; i++)
    {
        g << "Nodul: " << i << '\n';
        for(int j = 0; j < 6; j++)
            g << dp[i][j] << " ";
        g << '\n';
    } */
    for(int i = 0; i < m; i++)
    {
        f >> x >> p;
        g << query(x, p);
        g << '\n';
    }

}