91超碰碰碰碰久久久久久综合_超碰av人澡人澡人澡人澡人掠_国产黄大片在线观看画质优化_txt小说免费全本

溫馨提示×

Python TreeNode如何實現樹的AVL樹

小樊
83
2024-08-17 16:30:42
欄目: 編程語言

在Python中實現AVL樹,需要定義一個TreeNode類來表示樹的節點,然后定義一個AVLTree類來實現AVL樹的操作。

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

class AVLTree:
    def __init__(self):
        self.root = None

    def insert(self, key):
        self.root = self._insert(self.root, key)

    def _insert(self, node, key):
        if not node:
            return TreeNode(key)
        
        if key < node.key:
            node.left = self._insert(node.left, key)
        else:
            node.right = self._insert(node.right, key)

        node.height = 1 + max(self._get_height(node.left), self._get_height(node.right))
        
        balance = self._get_balance(node)

        # 左旋
        if balance > 1 and key < node.left.key:
            return self._rotate_right(node)
        # 右旋
        if balance < -1 and key > node.right.key:
            return self._rotate_left(node)
        # 左右旋
        if balance > 1 and key > node.left.key:
            node.left = self._rotate_left(node.left)
            return self._rotate_right(node)
        # 右左旋
        if balance < -1 and key < node.right.key:
            node.right = self._rotate_right(node.right)
            return self._rotate_left(node)

        return node

    def _get_height(self, node):
        if not node:
            return 0
        return node.height

    def _get_balance(self, node):
        if not node:
            return 0
        return self._get_height(node.left) - self._get_height(node.right)

    def _rotate_left(self, z):
        y = z.right
        T2 = y.left

        y.left = z
        z.right = T2

        z.height = 1 + max(self._get_height(z.left), self._get_height(z.right))
        y.height = 1 + max(self._get_height(y.left), self._get_height(y.right))

        return y

    def _rotate_right(self, z):
        y = z.left
        T3 = y.right

        y.right = z
        z.left = T3

        z.height = 1 + max(self._get_height(z.left), self._get_height(z.right))
        y.height = 1 + max(self._get_height(y.left), self._get_height(y.right))

        return y

以上是AVL樹的Python實現代碼,包括了插入節點和旋轉操作。可以根據需要進一步實現刪除節點、查找節點等操作。

0
衡水市| 聂拉木县| 泸定县| 左云县| 旬阳县| 汤阴县| 家居| 思南县| 雅安市| 正安县| 霍山县| 安宁市| 和林格尔县| 民乐县| 若尔盖县| 淮南市| 双峰县| 湟源县| 华安县| 夹江县| 同德县| 葫芦岛市| 乳山市| 宝鸡市| 莫力| 晋州市| 巴林左旗| 高碑店市| 涿州市| 惠水县| 宣恩县| 罗田县| 肥东县| 奈曼旗| 叶城县| 龙川县| 确山县| 桦南县| 盈江县| 苍山县| 修文县|