Construct Binary Tree from Preorder and Inorder Traversal

最新推荐文章于 2019-02-14 17:21:04 发布

原创最新推荐文章于 2019-02-14 17:21:04 发布 · 2k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Tree

本文详细介绍了如何通过给定的二叉树的前序和中序遍历序列，构建原始的二叉树结构。通过定义两个辅助函数，并在主函数中整合这些功能，实现了一个简洁且高效的算法。讨论了常见的错误处理，如输入序列长度不匹配等问题，并提供了优化内存使用的建议。

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

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

class Solution {
public:
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if (preorder.size() <= 0)
            return NULL;
            
        if (preorder.size() != inorder.size())
            return NULL;
            
        TreeNode* root = new TreeNode(0);
        root = buildTree(preorder, 0, preorder.size()-1, inorder,0, preorder.size()-1);
        
        return root;
    }
    
    
    TreeNode* buildTree(vector<int> &preorder, int preStart, int preEnd, vector<int> &inorder, int inStart, int inEnd){
        if (preEnd < preStart )
            return NULL;
            
        TreeNode* root = new TreeNode(0);
        root->val = preorder.at(preStart);
        if (preEnd== preStart)    
    		return root;
            
        int i;
        for(i = 0; i < inorder.size(); i++){
            if (inorder.at(inStart+i) == root->val)
                break;
        }
        
        root->left  = buildTree(preorder, preStart+1, preStart+i,inorder,inStart, inStart + i-1);
        root->right = buildTree(preorder, preStart+1+i, preEnd,inorder,inStart+ i+1, inEnd);
        return root;
    }
};

1. 最开始打算利用vector<int> c(begin, end)来复制每次所需要的子vector，但是发现当子树是空的时候，会出现run error。

注意vector<int> c(begin, end) 是复制 [begin,end)的内容，不包括end！！！

2. 改用push_back, 代码的确很简单，但是大数量时候占用太多memory。

3, 最后才改成标注起始，结束位置。