从前序与中序遍历序列构造二叉树Python解法
生活随笔
收集整理的這篇文章主要介紹了
从前序与中序遍历序列构造二叉树Python解法
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
給定兩個整數數組 preorder 和 inorder ,其中 preorder 是二叉樹的先序遍歷, inorder 是同一棵樹的中序遍歷,請構造二叉樹并返回其根節點。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal
?
例:
輸入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
輸出: [3,9,20,null,null,15,7]
解析:
使用遞歸調用,每次遞歸存儲當前子樹的根節點。先序遍歷的第一個即為根節點。將先序遍歷的第一個節點在中序遍歷中找出,在中序遍歷中此節點的左邊為左子樹,右邊為右子樹。記錄左子樹的長度L_left,先序遍歷中第一個節點(也就是根節點)向后L_left長度即為左子樹,然后剩余的為右子樹。
# Definition for a binary tree node. # class TreeNode(object): # def __init__(self, val=0, left=None, right=None): # self.val = val # self.left = left # self.right = right class Solution(object):def buildTree(self, preorder, inorder):""":type preorder: List[int]:type inorder: List[int]:rtype: TreeNode"""def build(pre_left, pre_right, mid_left, mid_right): # 遞歸函數if pre_left > pre_right or mid_left > mid_right: # 終止條件return Noneroot = TreeNode(preorder[pre_left]) # 存儲根節點mid_root = mid_leftwhile mid_root <= mid_right and inorder[mid_root] != preorder[pre_left]:mid_root += 1 # 尋找中序遍歷中的根節點diff = mid_root - mid_left # 左子樹長度root.left = build(pre_left + 1, pre_left + diff, mid_left, mid_root - 1) #左子樹root.right = build(pre_left + 1 + diff, pre_right, mid_root + 1, mid_right) # 右子樹return root # 返回根節點return build(0, len(preorder) - 1, 0, len(inorder) - 1)總結
以上是生活随笔為你收集整理的从前序与中序遍历序列构造二叉树Python解法的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: airpods电池容量(Airpods)
- 下一篇: PPT 标题文字排版思路分享