Cod sursa(job #1728295)

Utilizator benisBenis Mimimus benis Data 12 iulie 2016 18:08:39
Problema Stramosi Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.29 kb
#include <iostream>
#include <fstream>

using namespace std;

#define NMAX 250000
#define MMAX 300000
#define PMAX 10

//#define _DEBUG

int N, M, PMax = 0;
int Q[MMAX];
int P[MMAX];
int R[MMAX];

int stramosi[NMAX+1][PMAX+1];

int main(int argc, const char * argv[]) {
    int i, j;
    fstream fin("stramosi.in", fstream::in);
    fin >> N >> M;
    for (i = 1; i <= N; i++) {
        fin >> stramosi[i][1];
    }
    for (i = 0; i < M; i++) {
        fin >> Q[i] >> P[i];
        if (P[i] > PMax)
            PMax = P[i];
    }
    PMax = (PMax + N - 1) / N;
    if (PMax > PMAX)
        PMax = PMAX;
    for (j = 2; j <= PMax; j++) {
        for (i = 1; i <= N; i++) {
            int s = stramosi[i][j-1];
            if (s != 0)
                s = stramosi[s][1];
            stramosi[i][j] = s != 0 ? stramosi[s][1] : 0;
        }
    }
    
    for (i = 0; i < M; i++) {
        int p = P[i];
        int q = Q[i];
        while (p > PMax) {
            q = stramosi[q][PMax];
            p -= PMax;
        }
        R[i] = stramosi[q][p];
    }
    fin.close();
    
    
#ifdef _DEBUG
    for (int i = 0; i < M; i++)
        cout << R[i] << "\n";
#else
    fstream fout("stramosi.out", fstream::out);
    for (int i = 0; i < M; i++)
        fout << R[i] << "\n";
    fout.close();
#endif
    return 0;
}