I-ti nivo binarnog pretraživačkog stabla

Napisati funkciju koja:

int stablo_broj_cvorova_na_nivou(Cvor* koren, int nivo);
void stablo_stampaj_elemente_na_nivou(Cvor* koren, int nivo);
int stablo_suma_elemenata_na_nivou(Cvor* koren, int nivo);

Ulaz

Sa standardnog ulaza se unosi traženi broj $i$, a zatim, sve do kraja ulaza (EOF), niz celih brojeva koji se ubacuju u binarno pretraživačko stablo.

Izlaz

Na standardni izlaz ispisati broj čvorova na $i$-tom nivou, zatim sve elemente na $i$-tom nivou u rastućem redosledu, odvojene jednim razmakom, i na kraju sumu svih elemenata na $i$-tom nivou. Svaki rezultat treba ispisati u posebnom redu.

Primer

stdin

2
5 3 8 1 4 7 9

stdout

4
1 4 7 9
21

Primer

stdin

1
10 20 5 15 30

stdout

2
5 20
25

Primer

stdin

3
1 2 3 4 5

stdout

1
4
4

Rešenje

stablo.h

#ifndef STABLO_H
#define STABLO_H

#include <stddef.h>

typedef struct Cvor {
    int vrednost;
    struct Cvor* levo;
    struct Cvor* desno;
} Cvor;

/* Ubacivanje */
int stablo_ubaci(Cvor** koren, int x);

/* Uništavanje */
void stablo_unisti(Cvor* koren);

/* Obilasci */
void stablo_infiksno(Cvor* koren, void (*visit)(int));
void stablo_prefiksno(Cvor* koren, void (*visit)(int));
void stablo_postfiksno(Cvor* koren, void (*visit)(int));

/* Pretraga */
Cvor* stablo_pretrazi(Cvor* koren, int x);

/* Brisanje */
void stablo_obrisi(Cvor** koren, int x);

/* Analiza */
size_t stablo_velicina(Cvor* koren);
size_t stablo_visina(Cvor* koren);

#endif

main.c

#include <stdio.h>
#include <stdlib.h>
#include "stablo.h"

int stablo_broj_cvorova_na_nivou(Cvor* koren, int nivo)
{
	if (!koren || nivo < 0)
		return 0;

	if (nivo == 0)
		return 1;

	return stablo_broj_cvorova_na_nivou(koren->levo, nivo - 1) +
		   stablo_broj_cvorova_na_nivou(koren->desno, nivo - 1);
}

void stablo_ispisi_elemente_na_nivou(Cvor* koren, int nivo)
{
	if (!koren || nivo < 0)
		return;

	stablo_ispisi_elemente_na_nivou(koren->levo, nivo - 1);

	if (nivo == 0) {
		printf("%d ", koren->vrednost);
		return;
	}

	stablo_ispisi_elemente_na_nivou(koren->desno, nivo - 1);
}

int stablo_suma_elemenata_na_nivou(Cvor* koren, int nivo)
{
	if (!koren || nivo < 0)
		return 0;

	if (nivo == 0)
		return koren->vrednost;

	return stablo_suma_elemenata_na_nivou(koren->levo, nivo - 1) +
		   stablo_suma_elemenata_na_nivou(koren->desno, nivo - 1);
}

int main(void)
{
	Cvor *koren = NULL;

	int nivo;
	scanf("%d", &nivo);

	int x;
	while (scanf("%d", &x) != EOF) {
		stablo_ubaci(&koren, x);
	}

	printf("%d\n", stablo_broj_cvorova_na_nivou(koren, nivo));
	stablo_ispisi_elemente_na_nivou(koren, nivo);
	printf("\n");
	printf("%d\n", stablo_suma_elemenata_na_nivou(koren, nivo));

	stablo_unisti(koren);

	return 0;
}

stablo.c

#include "stablo.h"
#include <stdlib.h>

/* ---------- pomoćne ---------- */

static Cvor* cvor_novi(int x) {
    Cvor* n = (Cvor*)malloc(sizeof(Cvor));
    if (!n) return NULL;
    n->vrednost = x;
    n->levo = NULL;
    n->desno = NULL;
    return n;
}

/* ---------- uništavanje ---------- */

void stablo_unisti(Cvor* koren) {
    if (!koren) return;
    stablo_unisti(koren->levo);
    stablo_unisti(koren->desno);
    free(koren);
}

/* ---------- ubacivanje: BST ---------- */
/* Dogovor: x < vrednost ide levo, inače desno */

int stablo_ubaci(Cvor** koren, int x) {
    if (!koren) return 0;

    while (*koren) {
        if (x < (*koren)->vrednost)
            koren = &(*koren)->levo;
        else
            koren = &(*koren)->desno;
    }

    *koren = cvor_novi(x);
    return (*koren != NULL);
}

/* ---------- obilasci ---------- */

void stablo_infiksno(Cvor* koren, void (*visit)(int)) {
    if (!koren) return;
    stablo_infiksno(koren->levo, visit);
    visit(koren->vrednost);
    stablo_infiksno(koren->desno, visit);
}

void stablo_prefiksno(Cvor* koren, void (*visit)(int)) {
    if (!koren) return;
    visit(koren->vrednost);
    stablo_prefiksno(koren->levo, visit);
    stablo_prefiksno(koren->desno, visit);
}

void stablo_postfiksno(Cvor* koren, void (*visit)(int)) {
    if (!koren) return;
    stablo_postfiksno(koren->levo, visit);
    stablo_postfiksno(koren->desno, visit);
    visit(koren->vrednost);
}

Cvor *stablo_pretrazi(Cvor *koren, int x)
{
    if (!koren) return NULL;

    if (koren->vrednost == x)
        return koren;
    else if (x < koren->vrednost)
        return stablo_pretrazi(koren->levo, x);
    else
        return stablo_pretrazi(koren->desno, x);
}

void stablo_obrisi(Cvor **koren, int x)
{
    if (!koren || !*koren) return;

    if (x < (*koren)->vrednost) {
        stablo_obrisi(&(*koren)->levo, x);
        return;
    } 
    
    if (x > (*koren)->vrednost) {
        stablo_obrisi(&(*koren)->desno, x);
        return;
    }

    // Čvor za brisanje pronađen
    if (!(*koren)->levo) {
        Cvor *temp = (*koren)->desno;
        free(*koren);
        *koren = temp;
        return;
    }
    
    if (!(*koren)->desno) {
        Cvor *temp = (*koren)->levo;
        free(*koren);
        *koren = temp;
        return;
    }

    // Čvor sa dva deteta: nađi najmanji u desnom podstablu
    Cvor *parent = *koren;
    Cvor *successor = (*koren)->desno;
    while (successor->levo) {
        parent = successor;
        successor = successor->levo;
    }

    // Kopiraj vrednost naslednika
    (*koren)->vrednost = successor->vrednost;

    // Obriši naslednika
    if (parent->levo == successor)
        parent->levo = successor->desno;
    else
        parent->desno = successor->desno;
    free(successor);
}

/* ---------- size / height ---------- */

size_t stablo_velicina(Cvor* koren) {
    if (!koren) return 0;
    return 1
         + stablo_velicina(koren->levo)
         + stablo_velicina(koren->desno);
}

size_t stablo_visina(Cvor* koren) {
    if (!koren) return 0;
    size_t hl = stablo_visina(koren->levo);
    size_t hr = stablo_visina(koren->desno);
    return 1 + (hl > hr ? hl : hr);
}