Cod sursa(job #1968717)

Utilizator RobybrasovRobert Hangu Robybrasov Data 17 aprilie 2017 20:15:39
Problema Stramosi Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.57 kb
#include <cstdio>
#include <iostream>
#include <vector>
#include <bitset>
#define FOR(i, n) for (int i = 1; i <= n; ++i)
#define iter vector<int>::iterator
#define N 250001
#define S_MAX 1 << 23

using namespace std;

bitset<N> E;
vector<int> V[N];
int S[N], A[N], *L[N], D[N];
char STR[S_MAX];

void read(int &x, char* &p) {
    x = 0;
    for (; *p < '0' || *p > '9'; ++p);
    for (; *p >= '0' && *p <= '9'; ++p) {
        x = 10 * x + *p - '0';
    }
}

void dfs(int node) {
    E[node] = 1;
    S[++S[0]] = node;
    for (int *it = L[node]; *it != -1; ++it)
        dfs(*it);
    int len = S[0];
    for (int step = 1, k = 0; step <= len; step <<= 1, ++k) {
        V[node].push_back(S[len - step]);
    }
    --S[0];
}

int main() {
    FILE *f = fopen("stramosi.in", "r");
    FILE *g = fopen("stramosi.out", "w");
    fread(STR, 1, S_MAX, f);
    char *ptr = STR;
    int n, m;
    read(n, ptr);
    read(m, ptr);
    FOR(i, n) {
        int x;
        read(x, ptr);
        ++D[x];
        A[i] = x;
    }
    FOR(i, n) {
        L[i] = new int[D[i] + 1];
        L[i][D[i]] = -1;
        D[i] = 0;
    }
    FOR(i, n) {
        if (A[i])
            L[A[i]][D[A[i]]++] = i;
    }

    FOR(i, n) {
        if (!E[i]) {
            S[0] = 0;
            dfs(i);
        }
    }

    while (m--) {
        int node, p;
        read(node, ptr);
        read(p, ptr);
        int step = 1;
        for (int k = 0; k < V[node].size() && step <= p; step <<= 1, ++k)
            if (step & p)
                node = V[node][k];
        fprintf(g, "%d\n", (step > p ? node : 0));
    }
    
    return 0;
}