Într-un arbore cu rădăcină, spunem că rădăcina este pe nivelul 1, fiii rădăcinii pe nivelul 2, fiii fiilor rădăcinii pe nivelul 3, etc.

Cerința

Se dă vectorul de tați al unui arbore cu rădăcină cu n noduri și o valoare k. Determinați nodurile situate pe nivelul k în arbore.

Date de intrare

Fișierul de intrare knivel.in conține pe prima linie numărul de noduri n și valoarea k. Pe linia următoare se află vectorul de tați al arborelui, valorile fiind separate prin spații.

Date de ieșire

Fișierul de ieșire knivel.out va conține pe prima linie, în ordine crescătoare, nodurile situate pe nivelul k în arbore.

Restricții și precizări

• 1 ≤ n ≤ 100
• în vectorul de tați rădăcina este marcată cu 0
• pe nivelul k va exista cel puțin un nod

Exemplu

knivel.in

8 3
4 3 0 3 2 1 2 1

knivel.out

1 5 7

#include <bits/stdc++.h>


using namespace std;

ifstream cin("knivel.in");
ofstream cout("knivel.out");

vector <int> G[101];
int n  , p , x , y , k , T[102] , P[101] , q , d[101] , maxi;

void dfs(int x)
{
    P[x] = 1;
    for(auto i:G[x])
        if(!P[i])
        {
            d[i] = d[x] + 1;
            dfs(i);
        }
}

int main()
{
    cin >> n >> k;
    for(int i = 1 ; i <= n ; i++)
    {
        cin >> T[i];
        if(T[i] != 0)
        {
            G[T[i]].push_back(i);
            G[i].push_back(T[i]);
        }
        if(T[i] == 0) p = i;

    }

    dfs(p);
    for(int i = 1 ; i <= n ; i++)
        if(d[i] + 1 == k) cout << i << " ";
}