#include <iostream>
using namespace std;

#include "Conjunto.h"

Conjunto::Nodo::Nodo(int elDato) : dato{elDato}, tallaIzquierdo{0}, izquierdo{nullptr}, derecho{nullptr} {
}

Conjunto::Conjunto() : raiz{nullptr} {
}
  
void Conjunto::insertar(int unDato) {
   insertar(unDato, raiz);
}

bool Conjunto::insertar(int unDato, Nodo * & n) {
   if (n == nullptr) {
      n = new Nodo(unDato);
      return true;
   } else if (unDato < n->dato)
      if (insertar(unDato, n->izquierdo)) {
	 n->tallaIzquierdo++;
	 return true;
      } else
	 return false;
   else if (unDato > n->dato)
      return insertar(unDato, n->derecho);
   else
      // No insertamos duplicados
      return false;
}

int Conjunto::minimoEnSubarbol(Nodo * n) const { // Sabiendo que n != nullptr
   if (n->izquierdo == nullptr)
      return n->dato;
   return minimoEnSubarbol(n->izquierdo);
}

void Conjunto::eliminar(int unDato) {
   eliminar(unDato, raiz);
}

bool Conjunto::eliminar(int unDato, Nodo * & n) {
   if (n == nullptr)
      return false;
   else if (unDato < n->dato)
      if (eliminar(unDato, n->izquierdo)) {
	 n->tallaIzquierdo--;
	 return true;
      } else
	 return false;
   else if (unDato > n->dato)
      return eliminar(unDato, n->derecho);
   else if (n->izquierdo != nullptr && n->derecho != nullptr) {
      n->dato = minimoEnSubarbol(n->derecho);
      return eliminar(n->dato, n->derecho);
   } else {
      Nodo * basura = n;
      if (n->izquierdo != nullptr)
	 n = n->izquierdo;
      else
	 n = n->derecho;
      delete basura;
      return true;
   }
}

bool Conjunto::buscar(int unDato) const {
   Nodo * n = raiz;
   while (n != nullptr) {
      if (unDato < n->dato)
	 n = n->izquierdo;
      else if (unDato > n->dato)
	 n = n->derecho;
      else
	 return true;
   }
   return false;
}

int Conjunto::buscarKesimo(int k) const {
   if (k < 0)
      throw string("Usando buscarKesimo con un valor de k que no existe");
   return buscarKesimo(k, raiz);
}

int Conjunto::buscarKesimo(int k, Nodo * n) const {
   if (n == nullptr) // Si tenemos laTalla, esto se podria sustituir por
      // comprobar que k < laTalla en el metodo anterior
      throw string("Usando buscarKesimo con un valor de k que no existe");
   else if (k < n->tallaIzquierdo)
      return buscarKesimo(k, n->izquierdo);
   else if (k > n->tallaIzquierdo)
      return buscarKesimo(k - (n->tallaIzquierdo + 1), n->derecho);
   else
      return n->dato;
}