重建二叉树

最新推荐文章于 2024-08-31 20:59:22 发布

涛起云永

最新推荐文章于 2024-08-31 20:59:22 发布

阅读量42

点赞数

分类专栏：算法文章标签：算法数据结构 leetcode

本文链接：https://blog.csdn.net/qq_52399817/article/details/129665827

版权

算法专栏收录该内容

8 篇文章 0 订阅

订阅专栏

重建二叉树

问题描述

给定节点数为 n 的二叉树的前序遍历和中序遍历结果，请重建出该二叉树并返回它的头结点。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建出如下图所示。
在这里插入图片描述

提示:

1.vin.length == pre.length
2.pre 和 vin 均无重复元素
3.vin出现的元素均出现在 pre里

思路分析

先序遍历特点：每次读取的第一个值一定为根节点
中序遍历特点：根结点左边的值为左子树，右边为右子树

在这里插入图片描述
因此，我们每次都需要在前序遍历中找根节点并创建一个根节点，然后在中序遍历中确定根节点位置，并确定当前根节点的左右子树，然后以同样的方法去构建左右子树。这整个过程其实是一个递归的过程。

代码实现

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
#include <utility>
#include <vector>
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
    	//递归结束条件设定
         if(pre.empty()|| vin.empty()){
         return nullptr;
         }
         //记录根节点的位置
        int mid = 0;
        //寻找根节点在中序中的位置
        for(mid = 0; mid < vin.size(); ++mid){
            if(vin[mid] == pre[0])
            break;
        }
        //创建根节点
        TreeNode* root = new TreeNode(vin[mid]);
        vector<int>pre_left,pre_right;
        vector<int>vin_left,vin_right;
        //记录先序、中序根节点的左子树
        for(int i = 0; i < mid; ++i){
            pre_left.push_back(pre[i+1]);
            vin_left.push_back(vin[i]);
        }
		//记录先序、中序根节点的右子树
        for(int j = mid + 1; j < vin.size(); ++j){
            pre_right.push_back(pre[j]);
            vin_right.push_back(vin[j]);
        }

        //递归构造左右子树
        root->left = reConstructBinaryTree(pre_left,vin_left);
        root->right = reConstructBinaryTree(pre_right,vin_right);
        return root;
    }
};