Cod sursa(job #1756328)

Utilizator KronSabau Valeriu Kron Data 12 septembrie 2016 17:15:26
Problema Stramosi Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.7 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <iterator>
using namespace std;

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

int n,m;
vector <int> a[250001];
int main()
{
    int x;
    f >> n >> m;
    for(int i=1;i<=n;i++)
    {
        f >> x;
        if(x){
            a[i].push_back(x);
            for(vector<int>::iterator it=a[x].begin();it!=a[x].end();it++)
            {
                a[i].push_back(*it);
            }
        }
    }

    int k;
    for(int i=1;i<=m;i++)
    {

        f>> x >> k;

        if(a[x].size()>=k)
        {

            g<< a[x][k-1] << "\n";

        }else  g <<  "0\n";
    }

    return 0;
}