Cod sursa(job #3234015)

Utilizator Cristi1123Lucan Crisitian Cristi1123 Data 5 iunie 2024 21:32:18
Problema Stramosi Scor 70
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <iostream>
#include <vector>
#include <stdint.h>
#include <fstream>
using namespace std;

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

    uint32_t nr_people, nr_queries;
    fin >> nr_people >> nr_queries;

    vector<uint32_t> descendents(nr_people + 1), query_res;
    query_res.reserve(nr_queries);
    for (uint32_t i = 1; i <= nr_people; ++i)
        fin >> descendents[i];
    while (nr_queries--){
        uint32_t child, nr_descendents;
        fin >> child >> nr_descendents;
        
        while (nr_descendents-- && child)
            child = descendents[child];
        query_res.push_back(child);
    }

    for (const auto& res : query_res)
        fout << res << endl;
    
}