刷題系列 - 中序和後序遍歷佇列，構造對應二叉樹；

 

假期繼續刷題，也沒有別的什麼事情可以幹。

 

這個題是給出中序和後序遍歷佇列，構造對應二叉樹；題目很簡單，如下圖，給出兩個遍歷佇列，構成二叉樹，這裡假定沒有重複點。

 

想了好幾天，真是慚愧，因為一直想一次遍歷就完成構造，最後發現不行；然後就硬搞出一個多重迴圈的遍歷方法，雖然可行，但是提交後提示耗時超過限制。最後還是用遞迴實現的。

 

其實原理很簡單，對於後續遍歷佇列，最後一個值就是整個二叉樹的根節點；而這個根節點去掉後，可以把二叉樹分成左右兩個樹，在中序佇列中，按照這個根節點來拆分出可以得到左右佇列，分佈對應左邊樹和右邊樹的所有點。而且其實後序佇列也是按照左右樹節點劃分的，只要知道左右樹的節點數量，來劃分就可以了，這個可以從中序佇列劃分結果獲得。反覆同理，再劃分出來左右樹繼續劃分，可以得到葉子節點，或者空序列；這樣就完成樹的構成。

 

程式碼如下：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def buildTree(self, inorder: List[int], postorder: List[int]) -> TreeNode:
        if inorder == []:
            return None
        else:
            if len(inorder) == 1:
                return TreeNode(inorder[0])
            else:
                RootVal = postorder[-1]
                currentNode = TreeNode(RootVal)
                inorderLeft = inorder[:inorder.index(RootVal)]
                inorderRight = inorder[inorder.index(RootVal)+1:]
                postorder.pop()
                postorderLeft = postorder[:len(inorderLeft)]
                postorderRight = postorder[-len(inorderRight):] 
                currentNode.left = self.buildTree(inorderLeft,postorderLeft)
                currentNode.right = self.buildTree(inorderRight,postorderRight)
                return currentNode