当前位置:首页 > 开发语言/框架 > C++

[leetcode] construct-binary-tree-from-preorder-and-inorder

优良自学吧提供[leetcode] construct-binary-tree-from-preorder-and-inorder,[leetcode] construct-binary-tree-from-preorder-and-inorder-题目描述   Given preorder and inorder traversal of a tre

[leetcode] construct-binary-tree-from-preorder-and-inorder-

题目描述

 

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

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

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    
    
    void ConstructTree(TreeNode *&tree, vector<int> &preorder, int prebegin, int preend, vector<int> &inorder, int inbegin, int inend)
    {
        if(prebegin > preend || inbegin > inend)
            return;
        int i;
        int mid = preorder[prebegin];
        for(i = inbegin; i <= inend; i++)
        {
            if(inorder[i] == mid)
                break;
        }
        tree = new TreeNode(mid);
        ConstructTree(tree->left, preorder, prebegin+1, prebegin+i-inbegin, inorder, inbegin, i-1);
        ConstructTree(tree->right, preorder, prebegin+i+1-inbegin, preend, inorder, i+1, inend);
    }
    
    
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) 
    {
        if(preorder.size() == 0 || inorder.size() == 0)
            return NULL;
        TreeNode *tree = NULL;
        
        ConstructTree(tree, preorder, 0, preorder.size()-1, inorder, 0, inorder.size()-1);
        
        return tree;
    }
};

 


(本文来自互联网,不代表搜站(http://www.ylzx8.cn/)的观点和立场)
本站所有内容来自互联网,若本站收录的信息无意侵犯了贵司版权,请给我们来信(ylzx8cn@163.com),我们会及时处理和回复,谢谢