二叉树-前序遍历和中序遍历树构造二叉树-中等

描述

根据前序遍历和中序遍历树构造二叉树.

你可以假设树中不存在相同数值的节点

您在真实的面试中是否遇到过这个题？  是

样例

给出中序遍历：[1,2,3]和前序遍历：[2,1,3]. 返回如下的树:

  2
 / \
1   3

题目链接

程序

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param inorder: A list of integers that inorder traversal of a tree
     * @param postorder: A list of integers that postorder traversal of a tree
     * @return: Root of a tree
     */
    TreeNode * buildTree(vector<int> &preorder, vector<int> &inorder) {
        // write your code here
        TreeNode *root = NULL;  // 定义一个根节点
        //存放根节点左右两边的子节点
        vector<int> l_inorder, r_inorder, l_preorder, r_preorder;
        int i, root_index = 0;
        
        if(preorder.empty() != 1 || inorder.empty() != 1){//不是空，则存放头结点
            root = new TreeNode(preorder[0]);//  在前序队列中找根节点
            
            //  在中序队列中找出根节点位置
            for(i = 0; i < inorder.size(); i++){
                if(preorder[0] == inorder[i])
                    break;
                root_index++;
            }
            //  左子树的前序、中序队列
            for(i = 0; i < root_index; i++){
                l_preorder.push_back(preorder[i+1]);//0位置为根节点
                l_inorder.push_back(inorder[i]);//root_index位置为根节点
            }
            //  右子树的前序、中序队列
            for(i = root_index+1; i < inorder.size(); i++){//已过根节点位置
                r_preorder.push_back(preorder[i]);
                r_inorder.push_back(inorder[i]);
            }
            
            root->left = buildTree(l_preorder, l_inorder);
            root->right = buildTree(r_preorder, r_inorder);
        }
        
        return root;
    }
};