How do I implement an AVL tree in Python?

How do I implement an AVL tree in Python?

To implement an AVL tree in Python, you need to create a class for the tree and define methods for insertion, deletion, and balancing.

What are the key properties of an AVL tree?

An AVL tree is a self-balancing binary search tree where the difference in heights between left and right subtrees is at most one.

Can you show me a sample code for inserting a node?

Sure! Here's a simple example:

class Node:
    def __init__(self, key):
        self.key = key
        self.left = None
        self.right = None
        self.height = 1

class AVLTree:
    def insert(self, root, key):
        # Normal BST insert
        if not root:
            return Node(key)
        elif key < root.key:
            root.left = self.insert(root.left, key)
        else:
            root.right = self.insert(root.right, key)

        # Update height and balance the tree
        return self.balance(root)

How do I balance the tree after insertion?

You can balance the tree by checking the balance factor of each node and performing rotations (left or right) as needed.