Cod sursa(job #790026)

Utilizator andrei.sfrentSfrent Andrei andrei.sfrent Data 20 septembrie 2012 01:05:00
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <cstdio>

int N, M, *S;

int log2(int x)
{
    int c = 0;
    while(x)
    {
        c++;
        x = x >> 1;
    }
    return c - 1;
}

void read()
{
    scanf("%d%d\n", &N, &M);
    S = new int[N + 1];
    S[0] = 0;
    for(int i = 1; i <= N; i++)
        scanf("%d", S + i);
}

#define NECALCULAT -1
#define MAXN 250002
int mem[33][MAXN];

void pre()
{
    for(int i = 0; i <= N; i++)
        mem[0][i] = S[i];
    for(int j = 1; j < log2(M) + 1; j++)
        for(int i = 0; i <= N; i++)
            mem[j][i] = mem[j - 1][mem[j - 1][i]];
}

int solve(int p, int q)
{
    int u = p & (p - 1);
    if(u == 0) // putere a lui 2
        return mem[log2(p)][q];
    return solve(u, mem[log2(p - u)][q]);
}

int main()
{
    freopen("stramosi.in", "r", stdin);
    freopen("stramosi.out", "w", stdout);
    read();
    pre();
    int p, q;
    while(M--)
    {
        scanf("%d%d", &q, &p);
        printf("%d\n", solve(p, q));
    }
    return 0;
}