Cerința

Se dă vectorul de tați al unui arbore cu rădăcină cu n noduri și un nod k. Afișați, în ordine crescătoare, nodurile terminale din subarborele cu rădăcina în k.

Date de intrare

Fișierul de intrare subarbore1.in conține pe prima linie numărul de noduri n și nodul 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 subarbore1.out va conține pe prima linie nodurile terminale din subarborele cu rădăcina în k, în ordine crescătoare, separate printr-un spațiu.

Restricții și precizări

• 1 ≤ n ≤ 100
• 1 ≤ k ≤ n
• în vectorul de tați rădăcina este marcată cu 0

Exemplu

subarbore1.in

8 4
4 3 0 3 2 1 4 1

subarbore1.out

6 7 8

#include <bits/stdc++.h>
using namespace std;

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

int n , k , t[101];

int main()
{
    cin >> n >> k;
    for(int i = 1 ; i <= n ; i++)
        cin >> t[i];

    for(int i = 1 ; i <= n ; i++)
    {
        int ok = 0;
        for(int j = 1 ; j <= n ; j++)
            if(t[j] == i) ok++;
        if(ok == 0)
        {
            int p = t[i];
            while(p)
            {
                if(p == k)ok = 1;
                p = t[p];
            }
            if(ok) cout << i << " ";
        }
    }
    return 0;

}