Cod sursa(job #2042300)

Utilizator ifrimencoAlexandru Ifrimenco ifrimenco Data 18 octombrie 2017 12:56:17
Problema Stramosi Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.83 kb
#include <iostream>
#include <fstream>

using namespace std;

#define NMAX 250005
ifstream f("stramosi.in");
ofstream g("stramosi.out");
int n;
int dp[20][NMAX];

int cautbin (int q, int p) {
    int step;
    if (p == 1) return dp[0][q];
    for (step = 1; (1 << step) <= p; ++step);
    step--;
    if (1 << step == p) return dp[step][q];
    int ans;
        for (int i = 0; step >= 0; step--) {
            q = dp[step][q];
        }
    return q;
}

int main()
{
    int m, i, j;
    f >> n >> m;
    for (i = 1; i <= n; ++i) {
        f >> dp[0][i];
    }
    for (i = 1; i <= 18; ++i)
        for (j = 1; j <= n; ++j)
            dp[i][j] = dp[i - 1][dp[i - 1][j]];

    int p, q;
    for (i = 1; i <= m; ++i) {
        f >> q >> p;
        g << cautbin(q, p) << "\n";

    }


    return 0;
}