Binarniy vyhladavaci strom [C]

Uvažujte binárny vyhľadávací strom. Na vstupe je daných niekoľko čísel, ktoré postupne vkladáte do tohto stromu. Pre každé číslo zistite, ako hlboko je v strome uložené. Ak sa číslo v strome nenachádza, tak ho najprv pridajte do stromu.

Špecifikácia vstupu: Na vstupe je niekoľko čísel, ktoré postupne vkladáte do stromu. Čísla sú oddelené medzerou alebo novým riadkom.

Špecifikácia výstupu: Pre každé číslo na vstupe vypíšte jedno číslo -- ako hlboko je dané číslo uložené v strome.

5
3
2
1
3

0
1
2
3
1

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

typedef struct strom{
   int hodnota;
     int hlbka;
   struct strom *vacsi;
     struct strom *mensi;
}strom;

int main()
{
  int i,p_hlbka=1;
  strom *prvy, *pom;
 
  prvy = (strom*)malloc(sizeof(strom));
  pom = prvy;
  prvy->hlbka=0;
  prvy->vacsi=NULL;
  prvy->mensi=NULL;
 
  scanf("%d",&prvy->hodnota);
 
 
  printf("%d\n",prvy->hlbka);
 
  while((scanf("%d",&i)) == 1)
  {
    p_hlbka = 1;
    pom=prvy;
    while(pom->vacsi != NULL || pom->mensi != NULL)
    {
      if(i==pom->hodnota)
        break;
      else if(i<pom->hodnota)
        pom=pom->mensi;
      else if(i>pom->hodnota)
        pom=pom->vacsi;
       
      p_hlbka++;
    }
   
    if(i<pom->hodnota)
    {
      pom->mensi = (strom*)malloc(sizeof(strom));
      pom = pom->mensi;
      pom->hodnota = i;
      pom->hlbka = p_hlbka;
      pom->vacsi = NULL;
      pom->mensi = NULL;
    }
    if(i>pom->hodnota)
    {
      pom->vacsi = (strom*)malloc(sizeof(strom));
      pom = pom->vacsi;
      pom->hodnota = i;
      pom->hlbka = p_hlbka;
      pom->vacsi = NULL;
      pom->mensi = NULL;
    }
   
    printf("%d\n",pom->hlbka);
  }
 
  return 0;
}

while(pom->vacsi != NULL || pom->mensi != NULL)
    {
      if(i==pom->hodnota)
        break;
      else if(i<pom->hodnota)
        pom=pom->mensi;
      else if(i>pom->hodnota)
        pom=pom->vacsi;
       
      p_hlbka++;
    }

  ak mám ľavý, alebo pravý podstrom, tak:
    ak sa hodnota rovná, našiel som, OK
    ak je i menšie, idem do ľavého podstromu (Ale čo ak nemám ľavý podstrom?)
    ak je i väčšie, idem do pravého podstromu (Ale čo ak nemám práve pravý podstrom?)

while(1)
    {
      if(i == pom->hodnota) //nasiel som, koncim tu
        break;

      if(i < pom->hodnota) //ak je hodnota mensia
      {
        if( pom->mensi ) //a existuje lavy podstrom
            pom = pom->mensi; //idem do laveho podstromu
        else
            break; //inak koncim a pridam uzol
      }

      if(i > pom->hodnota) //ak je hodnota vacsia
      {
        if( pom->vacsi ) //a existuje pravy podstrom
            pom = pom->vacsi; //idem do praveho podstromu
        else
            break; //inak koncim a pridam uzol
      }

      p_hlbka++;
    }