Cod sursa(job #1990227)

Utilizator FlorinHajaFlorin Gabriel Haja FlorinHaja Data 10 iunie 2017 21:07:16
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <fstream>
#include <vector>

using namespace std;

ifstream f("stramosi.in");
ofstream g("stramosi.out");

const int nmax = 250005;
int n, m, i, j, x, y;
bool viz[nmax];
int stv[nmax];
int sol[300005];
vector <int> ls[nmax];
vector <int> qr[nmax], lv[nmax];

void dfs(int x, int niv) {
    int l = ls[x].size(), i, y, w;
    viz[x] = 1;
    for (i = 0; i < l; i++) {
        y = ls[x][i];
        stv[niv] = x;
        dfs(y,niv+1);
    }
    l = qr[x].size();
    for (i = 0; i < l; i++) {
        w = qr[x][i];
        y = lv[x][i];
        sol[w] = stv[niv-y];
    }
}

int main() {
    f >> n >> m;
    for (i = 1; i <= n; i++) {
        f >> x;
        if (x) ls[x].push_back(i);
    }

    for (i = 1; i <= m; i++) {
        f >> x >> y;
        qr[x].push_back(i);
        lv[x].push_back(y);
    }
    for (i = 1; i <= n; i++)
        if (viz[i] == 0)
            dfs(i, 1);

    for (i = 1; i <= m; i++)
        g << sol[i] << '\n';
}