快樂的LeetCode --- 劍指 Offer 07. 重建二叉樹

題目描述：

輸入某二叉樹的前序遍歷和中序遍歷的結果，請重建該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。

例如，給出

前序遍歷 preorder = [3,9,20,15,7]
中序遍歷 inorder = [9,3,15,20,7]
返回如下的二叉樹：

    3
   / \
  9  20
    /  \
   15   7

---

解題思路1:

已知二叉樹的前序與中序，由於前序遍歷的順序是：打印 - 左 - 右，中序遍歷的順序是：左 - 打印 - 右。

1. 首先根據前序遍歷已知根節點的位置，然後查找根節點在中序遍歷中的位置。
2. 在中序遍歷根節點的左邊，爲左子樹；中序遍歷根節點的右邊，爲右子樹；
3. 返回前序遍歷中，左右子樹在根節點首次出現的位置，爲下一層子樹的根節點，依次循環遍歷即可；

---

代碼：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def buildTree(self, preorder, inorder):
        if not preorder:
            return None
        #創建當前節點
        node = TreeNode(preorder[0]) 
        #查找當前的根節點在中序遍歷中的位置
        index = inorder.index(preorder[0]) 
        # 劃分左右子樹
        left_pre = preorder[1:index+1]
        left_in = inorder[:index]
        right_pre = preorder[index+1:]
        right_in = inorder[index+1:]
        # 遍歷創建子樹
        node.left = self.buildTree(left_pre, left_in)
        node.right = self.buildTree(right_pre, right_in)
        # 返回當前節點
        return node

---

題目來源：

https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof