Cod sursa(job #1968622)

Utilizator RobybrasovRobert Hangu Robybrasov Data 17 aprilie 2017 19:31:48
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.41 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> L[N], V[N], S;
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.push_back(node);
    for (iter it = L[node].begin(); it != L[node].end(); ++it)
        dfs(*it);
    int len = S.size();
    for (int step = 1, k = 0; step < len; step <<= 1, ++k) {
        V[node].push_back(S[len - step - 1]);
    }
    S.pop_back();
}

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);
        if (x)
            L[x].push_back(i);
    }
    
    FOR(i, n) {
        if (!E[i]) {
            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;
}