Cod sursa(job #2977238)

Utilizator stefan05Vasilache Stefan stefan05 Data 11 februarie 2023 09:50:44
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.6 kb
#include <fstream>

#define NMAX 250005
#define LOG 18

using namespace std;

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

int n, m;
int p, q;
int dp[NMAX][LOG + 2];
int i, j, t;

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

    for (j = 1; j <= LOG; ++j)
        for (i = 1; i <= n; ++i)
            dp[i][j] = dp[dp[i][j-1]][j-1];

    for (t = 1; t <= m; ++t)
    {
        fin >>q>>p;

        for (j = LOG; j >= 0; j--)
            if ((p>>j)&1) q = dp[q][j];

        fout <<q<<'\n';
    }
    fout.close();
    return 0;
}