feat: invertHEADmain

1 files changed, 25 insertions, 0 deletions

diff --git a/bst.cc b/bst.cc
index f5f4f07..7101a63 100644
--- a/bst.cc
+++ b/bst.cc
@@ -200,6 +200,25 @@ public:
 
     std::cout << '\n';
   }
+
+  static auto invert(std::optional<std::shared_ptr<bst>> bst)
+      -> std::optional<std::shared_ptr<class bst>> {
+    if (bst.has_value()) {
+      if (bst.value()->_left.has_value() || bst.value()->_right.has_value()) {
+        auto left = invert(bst.value()->_left);
+        auto right = invert(bst.value()->_right);
+
+        bst.value()->_left = right;
+        bst.value()->_right = left;
+      }
+
+      return bst;
+    }
+
+    return std::nullopt;
+  }
+
+  auto invert() -> void { invert(std::make_shared<bst<T>>(*this)); }
 };
 
 auto main() -> int {
@@ -227,5 +246,11 @@ auto main() -> int {
 
   std::cout << "size: " << bst.size() << '\n';
 
+  std::cout << "inverted: ";
+
+  bst.invert();
+  bst.invert();
+  bst.print_from_largest_to_smallest();
+
   return 0;
 }