summaryrefslogtreecommitdiff
path: root/src/main_gen.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/main_gen.rs')
-rw-r--r--src/main_gen.rs45
1 files changed, 45 insertions, 0 deletions
diff --git a/src/main_gen.rs b/src/main_gen.rs
new file mode 100644
index 0000000..b3f6add
--- /dev/null
+++ b/src/main_gen.rs
@@ -0,0 +1,45 @@
+use std::cmp::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
+ }
+}
+
+fn main() {
+ let mut tree = AVLTreeSet::new();
+ tree.insert(3);
+ tree.insert(2);
+}