Cod sursa(job #1673322)

Utilizator mariakKapros Maria mariak Data 3 aprilie 2016 17:34:19
Problema Stramosi Scor 100
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[Log][Nmax];
int main()
{
    int i, j, b, p, node;
    scanf("%d %d", &n, &m);

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

    return 0;
}