剑指offer——重建二叉树

最新推荐文章于 2022-03-30 09:35:27 发布

prettysky123

最新推荐文章于 2022-03-30 09:35:27 发布

阅读量96

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer

本文链接：https://blog.youkuaiyun.com/prettysky123/article/details/90054846

剑指offer 专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种基于递归思想的二叉树重建算法，利用前序遍历和中序遍历结果，实现二叉树的重建过程。通过确定根节点，并划分左右子树的遍历序列，递归构建整棵树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

思路：使用递归的思想，在当前pre,vin中。
先通过先序遍历的性质，pre第一个元素pre[0]为当前二叉树的根节点root。
然后在中序vin中找到root所在的位置，记为it。故在vin中，it以左的元素位于左子树，以右的元素归右子树，即leftvin, rightvin。
之后在先序pre中找出属于左子树的先序遍历序列和属于右子树的先序遍历序列，即leftpre,rightvin。根据leftvin.size() == leftpre.size() , rightvin.size() ==rightpre.size(),从pre中截取各自的序列。
注意：void assign(const_iterator first,const_iterator last);将区间 [first,last)（不包含last对应的元素） 的元素赋值到当前的vector容器中，或者赋n个值为x的元素到vector容器中，这个容器会清除掉vector容器中以前的内容。

/**
 * 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) {
            TreeNode* T = NULL;
            if(pre.size() == 0 || vin.size() == 0)
                return T;
            int root = pre[0];
            T = new TreeNode(root);


            vector <int> :: iterator it = find(vin.begin(), vin.end(), root);

            vector <int> leftvin;
            leftvin.assign(vin.begin(), it);
            vector <int> rightvin;
            rightvin.assign(it+1, vin.end());

            vector <int> leftpre;
            vector <int> :: iterator leftprebegin =  pre.begin() + 1;
            vector <int> :: iterator leftpreend = pre.begin() + leftvin.size() + 1;
            leftpre.assign(leftprebegin, leftpreend);

            vector <int> rightpre;
            vector <int> :: iterator rightprebegin =  leftpreend;
            vector <int> :: iterator rightpreend = pre.end();

            rightpre.assign(rightprebegin, rightpreend);

            T->left = reConstructBinaryTree(leftpre, leftvin);
            T->right = reConstructBinaryTree(rightpre, rightvin);

            return T;

        
    }
};