Cod sursa(job #2340753)

Utilizator andra_moldovanAndra Moldovan andra_moldovan Data 10 februarie 2019 21:44:20
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <fstream>
#include <vector>
#include <queue>

#define MAXN 250002

using namespace std;

ifstream fin("stramosi.in");
ofstream fout("stramosi.out");

int N, M, viz[MAXN], sol[MAXN];

vector <int> graph[MAXN], s;
queue <int> Q;

struct str{
    int val, poz;
};

vector <str> v[MAXN];

inline void Read() {
    int x, y;

    fin >> N >> M;

    for (int i = 1; i <= N; i++) {
        fin >> x;
        if (!x) {
            Q.push(i);
        }
        else {
            graph[x].push_back(i);
        }
    }

    for (int i = 1; i <= M; i++){
        fin >> x >> y;

        v[x].push_back({y, i});
    }
}

inline void DFS(int node) {
    viz[node] = 1;
    if (!v[node].empty()) {
        int ll = s.size();
        for (auto j : v[node]) {
            if (ll - j.val >= 0)
                sol[j.poz] = s[ll - j.val];
        }
    }
    s.push_back(node);

    for (auto x : graph[node]) {
        if (!viz[x]) {
            DFS(x);
        }
    }
    s.pop_back();
 }

inline void Solve() {
    int z;
    while (!Q.empty()) {
        z = Q.front();
        Q.pop();
        if (viz[z])
            continue;
        s = vector <int> ();
        DFS(z);
    }
}

inline void Write() {
    for (int i = 1; i <= M; i++)
        fout << sol[i] << "\n";
}

int main () {
    Read();
    Solve();
    Write();

    fin.close(); fout.close(); return 0;
}