Cod sursa(job #3224173)

Utilizator Toni07Stoica Victor Toni07 Data 14 aprilie 2024 21:01:47
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.16 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

const int Nmax = 250005;
const int Mmax = 300005;

vector<int> L[Nmax];
vector<pair<int, int>> queries[Nmax];
vector<int> currPath;
int sol[Mmax];
// queries[nod] = [{k, indice}] => voi rezolva queriul original
// (nod, k) si voi pune rezultatul in sol[indice]

void dfs(int node) {
    currPath.push_back(node);
    for(auto qry : queries[node]) {
        int idx = currPath.size() - 1 - qry.first;
        cout << idx << " ";
        if(idx >= 0) {
            sol[qry.second] = currPath[idx];
        }

    }
    for(int son : L[node]) {
        dfs(son);
    }
    currPath.pop_back();
}

int main() {
    int n, m;
    ifstream fin("stramosi.in");
    ofstream fout("stramosi.out");
    fin >> n >> m;
    for(int i = 1; i <= n; i++) {
        int tata;
        fin >> tata;
        L[tata].push_back(i);
    }
    // Offline:
    for(int i = 1; i <= m; i++) {
        int x, k;
        fin >> x >> k;
        queries[x].push_back({k, i});
    }
    dfs(0);
    for(int i = 1; i <= m; i++) {
        fout << sol[i] << "\n";
    }
    return 0;
}