Cod sursa(job #1673294)

Utilizator mariakKapros Maria mariak Data 3 aprilie 2016 17:04:38
Problema Stramosi Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <bits/stdc++.h>
#define Nmax 250005
#define Log 18
FILE *fin  = freopen("stramosi.in", "r", stdin);
FILE *fout = freopen("stramosi.out", "w", stdout);

using namespace std;
int n, m, A[Nmax][Log];
int main()
{
    int i, j, b, p, node;
    scanf("%d %d", &n, &m);

    for(j = 0; (1 << j) <= n; ++ j)
        for(i = 1; i <= n; ++ i)
        {
            if(j == 0)
                scanf("%d", &A[i][0]);
            else if(A[i][j - 1])
                A[i][j] = A[A[i][j - 1]][j - 1];
        }
    while(m --)
    {
        scanf("%d %d", &node, &p);
        for(b = 0; (1 << b) <= p; ++ b)
            if(p & (1 << b))
            {
                node = A[node][b];
                if(node == 0)
                    break;
            }
        printf("%d\n", node);
    }

    return 0;
}