Cod sursa(job #1990230)

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

using namespace std;

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

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

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

        dfs(y);
    }
    l = qr[x].size();
    for (i = 0; i < l; i++) {
        w = qr[x][i];
        y = lv[x][i];
        sol[w] = stv[niv[x]-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);

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