中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

python二叉樹遍歷代碼怎么寫

小億
85
2023-12-14 17:39:01
欄目: 編程語言

在Python中,可以使用遞歸或者迭代的方式來實現二叉樹的遍歷。以下是常見的三種二叉樹遍歷方式的代碼示例:

  1. 前序遍歷(Preorder Traversal):根節點 -> 左子樹 -> 右子樹

遞歸實現:

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

def preorderTraversal(root):
    if root is None:
        return []
    result = [root.val]
    result += preorderTraversal(root.left)
    result += preorderTraversal(root.right)
    return result

迭代實現:

def preorderTraversal(root):
    if root is None:
        return []
    stack = [root]
    result = []
    while stack:
        node = stack.pop()
        result.append(node.val)
        if node.right:
            stack.append(node.right)
        if node.left:
            stack.append(node.left)
    return result
  1. 中序遍歷(Inorder Traversal):左子樹 -> 根節點 -> 右子樹

遞歸實現:

def inorderTraversal(root):
    if root is None:
        return []
    result = []
    result += inorderTraversal(root.left)
    result.append(root.val)
    result += inorderTraversal(root.right)
    return result

迭代實現:

def inorderTraversal(root):
    if root is None:
        return []
    stack = []
    result = []
    while stack or root:
        if root:
            stack.append(root)
            root = root.left
        else:
            node = stack.pop()
            result.append(node.val)
            root = node.right
    return result
  1. 后序遍歷(Postorder Traversal):左子樹 -> 右子樹 -> 根節點

遞歸實現:

def postorderTraversal(root):
    if root is None:
        return []
    result = []
    result += postorderTraversal(root.left)
    result += postorderTraversal(root.right)
    result.append(root.val)
    return result

迭代實現:

def postorderTraversal(root):
    if root is None:
        return []
    stack = [root]
    result = []
    while stack:
        node = stack.pop()
        result.append(node.val)
        if node.left:
            stack.append(node.left)
        if node.right:
            stack.append(node.right)
    return result[::-1]

以上是常見的三種二叉樹遍歷方式的代碼示例。根據具體需求選擇合適的遍歷方式。

0
交城县| 烟台市| 佛山市| 富源县| 镇坪县| 梁河县| 蕉岭县| 青冈县| 长寿区| 北辰区| 淮南市| 海南省| 沐川县| 连山| 茌平县| 石棉县| 万荣县| 靖安县| 简阳市| 卢龙县| 南乐县| 仪征市| 青田县| 浦东新区| 邯郸县| 温宿县| 赫章县| 桑植县| 怀来县| 施秉县| 高密市| 鄱阳县| 巴马| 常山县| 山西省| 正阳县| 涿州市| 舞阳县| 华阴市| 鸡西市| 枝江市|