Cod sursa(job #3322858)

Utilizator pepsiM4A1Ozturk Arif pepsiM4A1 Data 15 noiembrie 2025 22:56:18
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include<cstdio>
using namespace std;

const int DIM=10000;
int poz=DIM-1;
char buff[DIM];
void citeste(int &nr)
{
    while(!('0'<=buff[poz]&&buff[poz]<='9'))
    {
        if(++poz==DIM)
        {
            fread(buff,1,DIM,stdin);
            poz=0;
        }
    }
    while(('0'<=buff[poz]&&buff[poz]<='9'))
    {
        nr=nr*10 + buff[poz]-'0';
        if(++poz==DIM)
        {
            fread(buff,1,DIM,stdin);
            poz=0;
        }
    }
}

const int NMAX = 250055;
const int LOG_MAX = 20;
int stramosi[NMAX][LOG_MAX + 10]; // stramosi[i][j] = al 2^j-lea stramos al nodului i;

void ComputeStramosi(int n) {
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= LOG_MAX; j++) {
            stramosi[i][j] = stramosi[stramosi[i][j - 1]][j - 1];
        }
    }
}
int MSB(int x) {
    return 31 - __builtin_clz(x);
}
int query(int q, int p) {
    for(int i = 20; i >= 0; --i) { // merg de la cel mai mare bit la cel mai mic
        if(p & (1 << i)) // verific daca bitul i este setat in p
            q = stramosi[q][i]; // daca bitul i este setat in p, urc cu 2^i pasi in arbore
    }
    return q;
}

int main() {
    freopen("stramosi.in", "r", stdin);
    freopen("stramosi.out", "w", stdout);

    int n = 0, queries = 0;
    citeste(n), citeste(queries);

    for (int i = 1; i <= n; i++) {
        citeste(stramosi[i][0]);
    }

    ComputeStramosi(n);

    for (; queries > 0; queries--) {
        int q = 0, p = 0;
        citeste(q), citeste(p);
        printf("%d\n", query(q, p));
    }

    return 0;
}