From d6dce7238ad59f67b6b9bad1b3e50984bb69e84e Mon Sep 17 00:00:00 2001
From: Raito Bezarius
Date: Mon, 25 Mar 2024 11:43:20 +0100
Subject: Initial commit

Signed-off-by: Raito Bezarius <masterancpp@gmail.com>
---
 src/main.rs | 61 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 src/main.rs

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..d5bb8f9
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,61 @@
+use std::cmp::Ordering;
+
+trait Ord {
+    fn cmp(&self, other: &Self) -> Ordering;
+}
+
+struct AVLNode<T: Ord> {
+    value: T,
+    left: AVLTree<T>,
+    right: AVLTree<T>,
+}
+
+type AVLTree<T> = Option<Box<AVLNode<T>>>;
+
+struct AVLTreeSet<T: Ord> {
+    root: AVLTree<T>,
+}
+
+impl<T: Ord> AVLTreeSet<T> {
+    pub fn new() -> Self {
+        Self { root: None }
+    }
+
+    pub fn insert(&mut self, value: T) -> bool {
+        let mut current_tree = &mut self.root;
+
+        while let Some(current_node) = current_tree {
+            match current_node.value.cmp(&value) {
+                Ordering::Less => current_tree = &mut current_node.right,
+                Ordering::Equal => return false,
+                Ordering::Greater => current_tree = &mut current_node.left,
+            }
+        }
+
+        *current_tree = Some(Box::new(AVLNode {
+            value,
+            left: None,
+            right: None,
+        }));
+
+        true
+    }
+}
+
+impl Ord for u32 {
+    fn cmp(&self, other: &Self) -> Ordering {
+        if *self < *other {
+            Ordering::Less
+        } else if *self == *other {
+            Ordering::Equal
+        } else {
+            Ordering::Greater
+        }
+    }
+}
+
+fn main() {
+    let mut tree = AVLTreeSet::new();
+    tree.insert(3);
+    tree.insert(2);
+}
-- 
cgit v1.2.3