[LeetCode]105 根据先序遍历和中序遍历构建二叉树

Construct Binary Tree from Preorder and Inorder Traversal(根据先序遍历和中序遍历构建二叉树)

【难度：Medium】
Given preorder and inorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.
给定一棵二叉树的先序遍历和中序遍历结果，重构该二叉树。
假定二叉树中无重复值出现。

---

解题思路

根据题意，我们先来看一棵二叉树的先序遍历和中序遍历的结果：

先序遍历：1，2，4，6，7，5，3，8，9
中序遍历：6，4，7，2，5，1，8，3，9
与LeetCode 106题类似，
由先序遍历的定义，我们知道第一个值必定是二叉树的根节点root。由而中序遍历的定义，我们又可以发现根节点是左右子树的分界点。结合两者和在106题的基础上，对106的代码稍作修改，由先序遍历来确定每一个根节点，然后在中序遍历中找到该节点的下标i，那么[start, i-1]就是该i节点的左子树，[i+1, end]就是该i节点的右子树，直到叶子节点的出现。

---

c++代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if (preorder.empty())
            return NULL;
        int root_index = 0;
        return createBST(preorder,inorder,root_index,preorder.size()-1,root_index);
    }
    TreeNode* createBST(vector<int>& preorder, vector<int>& inorder, int start, int end, int& index) {
        int v = preorder[index];
        int i = start;
        for (i; i <= end; i++)
            if (v == inorder[i])
                break;
        TreeNode* root = new TreeNode(v);
        if (i-1 >= start)
            root->left = createBST(preorder,inorder,start,i-1,++index);
        if (end >= i+1)
            root->right = createBST(preorder,inorder,i+1,end,++index);
        return root;
    }
};