C++詳解：劍指Offer 07. 重建二叉樹

原題LeetCode鏈接：劍指Offer 07. 重建二叉樹

題目：劍指Offer 07. 重建二叉樹

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

前序遍歷 preorder = [3,9,20,15,7]
中序遍歷 inorder = [9,3,15,20,7]

返回如下二叉樹

    3
   / \
  9  20
    /  \
   15   7

思路

​ 給出兩個遍歷序列來建二叉樹這種題目解決方法幾乎是一致的。

​ 1.前序遍歷序列的第一個元素是整棵樹的根節點

​ 2.在中序遍歷中找到根節點，它的左邊就是左子樹的元素，右邊就是右子樹的元素

​ 3.計算出左子樹和右子樹各有多少節點，然後繼續在先序遍歷中尋找左右子樹的根節點，遞歸執行下去，直到完成。

C++代碼

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        return build(preorder, inorder, 0, preorder.size()-1, 0, inorder.size()-1);
    }
    //類似建樹的題目都需要另外寫一個函數
    //preL,PreR,inL,inR分別是前序遍歷起始和結束位置、後序遍歷起始和結束位置
    TreeNode* build(vector<int>& preorder, vector<int>& inorder, int preL, int preR, int inL, int inR){
        if(preL > preR || inL > inR) return NULL;
        TreeNode* root = new TreeNode(preorder[preL]);
        int i;
        for(i = inL; i < inR; i++){
            if(preorder[preL] == inorder[i])
                break;
        }
        root->left = build(preorder, inorder, preL+1, preL + i - inL, inL, i-1);
        root->right = build(preorder, inorder, preL+i-inL+1, preR, i+1, inR);
        return root;
    }
};

複雜度分析

時間複雜度：O(n).每個結點都要經歷一遍建樹過程

空間複雜度：O(n).存儲整棵樹