Cod sursa(job #2293035)

Utilizator PinkiePie1189Preoteasa Mircea-Costin PinkiePie1189 Data 30 noiembrie 2018 14:13:32
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.2 kb
#include <stdio.h>
#define MAXN 250000
#define MAXLOGN 18


int AnswerMyQuery(int node, int index);

void calculateMyD(int N);

FILE *fin, *fout;


int tata[MAXN + 1];

int d[MAXLOGN + 1][MAXN + 1];

int main() {
    fin = fopen("stramosi.in", "r");
    fout = fopen("stramosi.out", "w");


    int N, Q;

    fscanf(fin, "%d%d", &N, &Q);

    for(int i = 0; i < N; i++) {
            fscanf(fin , "%d", &tata[i + 1]);
    }

    calculateMyD(N);

    for(int i = 0; i < Q; i++) {
        int node, index;
        fscanf(fin, "%d%d", &node, &index);
        fprintf(fout, "%d\n", AnswerMyQuery(node, index));
    }
    fclose(fin);
    fclose(fout);
    return 0;
}

int AnswerMyQuery(int node, int index) {

    int currnode = node;
    int level = 0;
    while(index > 0) {
        if(index % 2 == 1) {
            currnode = d[level][currnode];
        }
        level++;
        index /= 2;
    }
    return currnode;
}

void calculateMyD(int N) {

    for(int j = 1; j <= N; j++) {
        d[0][j] = tata[j];
    }
    for(int i = 1; i <= MAXLOGN; i++) {
        for(int j = 1; j <= N; j++) {
            d[i][j] = d[i - 1][d[i - 1][j]];
        }
    }
}