码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 105. Construct Binary Tree from Preorder and Inorder Traversal

时间:2018-06-22 22:46:40      阅读:233      评论:0      收藏:0      [点我收藏+]

标签:div   leetcode   hat   back   col   duplicate   sum   uil   tco   

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

Note:
You may assume that duplicates do not exist in the tree.

For example, given

preorder = [3,9,20,15,7]
inorder = [9,3,15,20,7]

Return the following binary tree:

    3
   /   9  20
    /     15   7


已知二叉树的前序遍历这中序遍历,求二叉树,这个其实挺简单的,在前序遍历中取第一个元素,然后在中序遍历找到相应的元素,左边的为左子树,右边的为右子树,根据长度在前序遍历中找到相应左子树和右子树的前序遍历。然后就可以递归了。

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        
        vector<int> leftpre;
        vector<int> leftino;
        vector<int> rightpre;
        vector<int> rightino;
        
        if(preorder.empty()||inorder.empty())
            return NULL;
        int root = preorder[0]; int left=0;int right=0;int tag=0;
        for(int i=0;i<inorder.size();i++)
        {
            if(inorder[i]==root)
            {
                tag=1;
            }
            else if(tag==0)
            {left++;leftino.push_back(inorder[i]);}
            else
            {right++;rightino.push_back(inorder[i]);}
        }
        for(int i=1;i<left+1;i++)
        {
            leftpre.push_back(preorder[i]);
        }
        for(int i=left+1;i<preorder.size();i++)
        {
            rightpre.push_back(preorder[i]);
        }
        
        TreeNode* tree = new TreeNode(root);
       
        tree->left = buildTree(leftpre,leftino);
        tree->right = buildTree(rightpre,rightino);
        
        return tree;
        
    }
};

 

LeetCode 105. Construct Binary Tree from Preorder and Inorder Traversal

标签:div   leetcode   hat   back   col   duplicate   sum   uil   tco   

原文地址:https://www.cnblogs.com/dacc123/p/9215539.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!