leetcode105. 从前序与中序遍历序列构造二叉树Construct BinaryTree From Preorder And Inorder Traversal

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

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

例如，给出

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

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

思路：

前序第一个是根节点，然后根据中序的根节点位置，可以将树变成左右两棵子树。

然后递归

1. 是否为空？
2. new申请根节点
3. 找到根节点的位置
4. 分为左右两个遍历序列，并且递归

/**
 * 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 NULL;      
        TreeNode* tree=new TreeNode(preorder[0]);//new申请根节点
        
        vector<int>::iterator it;//找到根节点的位置
        it=find(inorder.begin(),inorder.end(),preorder[0]);
        int position=0;
        if(it!=inorder.end())
            position=distance(inorder.begin(),it);
        
        if(position==1)//左序列
            tree->left=new TreeNode(inorder[0]);
        else{
            vector<int> left_pre(preorder.begin()+1,preorder.begin()+position+1);  
            vector<int> left_ino(inorder.begin(),inorder.begin()+position);
            tree->left=buildTree(left_pre,left_ino);//递归
        }
        
        if(preorder.size()-position==2)//右序列
             tree->right=new TreeNode(inorder[inorder.size()-1]);
        else{
            vector<int> right_pre(preorder.begin()+position+1,preorder.end());
            vector<int> right_ino(inorder.begin()+position+1,inorder.end());    
            tree->right=buildTree(right_pre,right_ino);//递归
        }
        return tree;
    }
};