Cod sursa(job #2640323)

Utilizator BAlexandruBorgovan Alexandru BAlexandru Data 6 august 2020 01:04:37
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <fstream>

using namespace std;

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

int n, m;

int stramos[250001][18];
int log2[250001];

int main()
{
    f >> n >> m;

    for (int i=1; i<=n; i++)
        f >> stramos[i][0];

    for (int i=2; i<=n; i++)
        log2[i] = log2[i/2] + 1;

    for (int i=1; (1<<i)<=n; i++)
        for (int j=1; j<=n; j++)
            stramos[j][i] = stramos[stramos[j][i-1]][i-1];

    for (int i=1; i<=m; i++)
    {
        int nod, nrStramos;
        f >> nod >> nrStramos;

        int L = log2[nrStramos];
        for (int k=L; k>=0; k--)
            if (nrStramos - (1<<k) >= 0)
            {
                nrStramos -= (1<<k);
                nod = stramos[nod][k];
                if (!nod)
                    break;
            }

        g << nod << "\n";
    }

    return 0;
}