Maksimalni zbir puta u binarnom stablu

Napisati funkciju koja pronalazi maksimalan mogući zbir vrednosti duž nekog puta u binarnom pretraživačkom stablu. Put je niz čvorova gde je svaki sledeći sused prethodnog. Put može početi i završiti se u bilo koja dva čvora, ali mora sadržati bar jedan čvor.

int stablo_maksimalni_zbir_puta(Cvor* koren);

Ulaz

Sa standardnog ulaza se unosi, sve do kraja ulaza (EOF), niz celih brojeva koji se ubacuju u binarno pretraživačko stablo.

Izlaz

Na standardni izlaz ispisati maksimalni mogući zbir vrednosti duž nekog puta u stablu. Ako je stablo prazno, ispisati 0.

Primer

stdin

5 3 8 1 4 7 9

stdout

29

Primer

stdin

-10 -20 -5 -1 -7

stdout

-1

Primer

stdin

-10 -20 5 2 7 6 9

stdout

23

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 "stablo.h"

/* Vrati maksimalni zbir puta u stablu od korena do listova 
   i ažurira maksimalni zbir puta ako je potrebno */
int stablo_maksimalni_zbir_puta_pomocna(Cvor* koren, int* maksimalni_zbir)
{
	if (!koren)
		return 0;

	int zbir_levo = stablo_maksimalni_zbir_puta_pomocna(koren->levo, maksimalni_zbir);
	int zbir_desno = stablo_maksimalni_zbir_puta_pomocna(koren->desno, maksimalni_zbir);

	int zbir = koren->vrednost + zbir_levo + zbir_desno;
	if (*maksimalni_zbir < zbir)
		*maksimalni_zbir = zbir;

	return koren->vrednost + (zbir_levo > zbir_desno ? zbir_levo : zbir_desno);
}

int stablo_maksimalni_zbir_puta(Cvor* koren)
{
	int maksimalni_zbir = 0;
	stablo_maksimalni_zbir_puta_pomocna(koren, &maksimalni_zbir);
	return maksimalni_zbir;
}

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

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

	printf("%d\n", stablo_maksimalni_zbir_puta(koren));

	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);
}