Cod sursa(job #2973412)

Utilizator StefanL2005Stefan Leustean StefanL2005 Data 31 ianuarie 2023 22:04:36
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.61 kb
#include <iostream>
#include <vector>
#include <fstream>
using namespace std;
ifstream in("stramosi.in");
ofstream out("stramosi.out");


int Stramos(int Q, int P, vector<int> &v)
{
    if (P == 0)
        return Q;
    if (v[Q] == -1)
        return -1;

    return Stramos(v[Q], P - 1, v);
}
int main()
{
    int n, m;
    in>> n >> m;

    vector<int> v(n);
    for (int i = 0; i < n; i++)
    {
        in>> v[i];
        v[i]--;
    }

    for (int t = 0; t < m; t++)
    {
        int Q, P;
        in>> Q >> P; Q--;

        out<< Stramos(Q, P, v) + 1 << "\n";
    }
    return 0;
}