LeetCode之Construct Binary Tree from Preorder and Inorder Traversal

Given preorder and inorder traversal of a tree, construct the binary tree.

题目意思：给定前序和中序遍历，构建一个二叉树

思路：假设前序遍历的起始位置startPre，结束位置endPre，中序遍历：startIn，endIn

由前序遍历和中序遍历，可以确定哪边是左子树，哪边是右子树。在前序遍历中对应中序遍历，首先确定树根，即前序遍历第一个，对应中序遍历中的第i个数，然后可以根据“根”在中序遍历的位置：startPre+i-startIn划分左右子树范围；构建左子树，对应前序遍历的（0，startPre+i-startIn）,同样确定它是不是在中序遍历的前半段，不是，就构建右子树，则是（startPre+i-startIn+1，end）；



/**

 * 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) {

        TreeNode *root=rebuild(preorder,0,preorder.size()-1,inorder,0,inorder.size()-1);

        return root;

    }

    TreeNode* rebuild(vector<int>& Pre,int startPre,int endPre,vector<int>& In,int startIn,int endIn){

        if(startPre>endPre||startIn>endIn)

            return NULL;

        TreeNode *root=new TreeNode(Pre[startPre]);

        for(int i=startIn;i<=endIn;i++){

            if(In[i]==Pre[startPre]){

                root->left=rebuild(Pre,startPre+1,startPre+i-startIn,In,startIn,i-1);

                root->right=rebuild(Pre,i-startIn+startPre+1,endPre,In,i+1,endIn);

            }

        }

        return root;

    }

};