重建二叉树

最新推荐文章于 2020-03-15 00:55:24 发布

疯癫小程序

最新推荐文章于 2020-03-15 00:55:24 发布

阅读量147

点赞数

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/helonSY/article/details/80867536

版权

算法专栏收录该内容

34 篇文章 0 订阅

订阅专栏

/*
输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。
*/

/**

 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
if(pre.size() == 0 || vin.size() == 0) return NULL ;
        
        TreeNode* pRet = new TreeNode(pre[0]) ;
        std::vector<int> leftPre, leftVin ;
        std::vector<int> rightPre, rightVin ;
        int nLeftVinCnt = 0 ;
        bool bRight = false ;
        for(std::size_t i = 0; i < vin.size(); ++i) {
            if(!bRight) {
                if(pre[0] == vin[i]) {
                bRight = true ;
                continue ;
                }
                leftVin.push_back(vin[i]) ;
                ++nLeftVinCnt ;// ++nLeftVinCnt
            }else {
                rightVin.push_back(vin[i]) ;
            }
        }
        
        for(std::size_t i = 1; i < pre.size(); ++i) {
            if(nLeftVinCnt > 0) {
                leftPre.push_back(pre[i]) ;
                nLeftVinCnt-- ;
            }else {
                rightPre.push_back(pre[i]) ;
            }
        }
        
        pRet->left = reConstructBinaryTree(leftPre, leftVin) ;
        pRet->right = reConstructBinaryTree(rightPre, rightVin) ;// pRet->right
        
        return pRet ;
    }
};