Cod sursa(job #3208211)

Utilizator Iustin2812Ion Iustin Ciprian Iustin2812 Data 28 februarie 2024 00:00:06
Problema Stramosi Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

vector <int> tata[250002];
vector <int> rez;
vector <pair <int ,int>> queries[300001];
int rezolv[300001];
int viz[250001];
int q,p;

void dfs (int node){
    
    for(auto i:queries[node]){
        rezolv[i.second]=rez[rez.size()-i.first];
    }
    
    viz[node]=1;
    rez.push_back(node);
    for(int i:tata[node]){
        if(viz[i]==0){
            dfs(i);
        }
    }
    rez.pop_back();
    
}

int main() {
    int n,m,a;
    fin>>n>>m;
    for(int i=1;i<=n;i++){
        fin>>a;
        tata[a].push_back(i);
    }
    
    for(int i=1;i<=m;i++){
        fin>>q>>p;
        queries[q].push_back({p,i});
    }
    dfs(0);
    for(int i=1;i<=m;i++){
        fout<<rezolv[i]<<endl;
    }
    return 0;
}