Cod sursa(job #2023045)

Utilizator Dan_RadulescuRadulescu Dan Dan_Radulescu Data 18 septembrie 2017 09:15:42
Problema Stramosi Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <bits/stdc++.h>
#define N 250001
#define M 300001
using namespace std;

FILE *f1 = fopen("stramosi.in", "r");
FILE *f2 = fopen("stramosi.out", "w");

int dp[20][N], n, m, a, i, j, p, q, power, index, aux;
int main()
{
    fscanf(f1, "%d%d", &n, &m);
    for (i = 1; i <= n; i++)
        fscanf(f1, "%d", &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]];

    for (i = 1; i <= m; i++)
    {
        fscanf(f1, "%d%d", &q, &p);
        power = 1; index = 0;
        while (power <= p)
        {
            power = power * 2;
            index++;
        }
        power = power / 2;
        index--;
        a = dp[index][q];
        while (a != 0 && power < p)
        {
            power++;
            a = dp[0][a];
        }
        fprintf(f2, "%d\n", a);
    }

    fclose(f1);
    fclose(f2);
    return 0;
}