码迷,mamicode.com
首页 > 编程语言 > 详细

算法之重建二叉树

时间:2019-03-13 12:09:29      阅读:162      评论:0      收藏:0      [点我收藏+]

标签:重建   结果   一个   左右   style   end   col   lse   int   

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

 //m,n 前序数组的起点和终点 i,j 中序数组的起点和终点
TreeNode * ConstructSub(vector<int>&preOrderVec,vector<int>&inOrderVec,int m,int n,int i,int j){
    int rootValue = preOrderVec[m];
    TreeNode * root = new TreeNode(rootValue);
    root->left=nullptr;
    root->right=nullptr;

    //边界, 左边或者右边只有一个元素的时候,并且前序和中序的值相等
    if(m==n && i==j){
        if(preOrderVec[m]==inOrderVec[i]){
            return root;
        }
        else{return nullptr;}
    }


    //找到左右两边
    //中序序列里的root的索引
    int rootInorderIndex=i;
    //往后开始找
    while (rootInorderIndex<n&&inOrderVec[rootInorderIndex]!=rootValue) {
        rootInorderIndex++;
    }
    //找到了,那就划分左右子树,然后递归
    int leftLength= rootInorderIndex-i;
    int leftPreorderEndIndex= m+leftLength;
    //存在左子树
    if(leftLength>0){
        root->left = ConstructSub(preOrderVec, inOrderVec, m+1, leftPreorderEndIndex,i,rootInorderIndex-1);
    }
    //存在右子树
    if(leftLength<n-m){
        root->right= ConstructSub(preOrderVec, inOrderVec, leftPreorderEndIndex+1, n, rootInorderIndex+1, j);
    }
    return root;
}

 

算法之重建二叉树

标签:重建   结果   一个   左右   style   end   col   lse   int   

原文地址:https://www.cnblogs.com/xiaonanxia/p/10522188.html

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