105. 从前序与中序遍历序列构造二叉树

题目

根据一棵树的前序遍历与中序遍历构造二叉树。

注意

注意:
你可以假设树中没有重复的元素。

例如，给出

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

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

思路

• 抓住前序和中序的两个特点。
• 前序的第一个元素是当前树的根节点。
• 中序的根节点元素，左区间为左子树，右区间为右子树。
• 把前序和中序数组不停划分递归，用left和right记录区间的起始和结尾index，当left > right时，说明了当前区间已经没有元素了，返回nullptr。

代码

/**
 * 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.size() == 0 ) return nullptr;

        return helper( preorder, 0, preorder.size() - 1, 
                       inorder, 0, inorder.size() - 1 );
    }

    TreeNode* helper( vector<int>& Pre, int pLeft, int pRight, 
                      vector<int>& In, int inLeft, int inRight )
    {
        if ( pLeft > pRight ) 
            return nullptr;

        TreeNode* node = new TreeNode( Pre[pLeft] );

        if ( pLeft == pRight )
            return node;

        int mid = inLeft;
        while ( In[mid] != Pre[pLeft] )
            ++mid;

        int size = mid - inLeft;   // pLeft之后到mid处共有几个数。例如7-5 = 2，5后面到7有6，7两个数。

        node->left = helper( Pre, pLeft+1, pLeft + size, In, inLeft, mid - 1 );
        node->right = helper( Pre, pLeft + size + 1, pRight, In, mid + 1, inRight );

        return node;
    }
};