LeetCode106. 从中序与后序遍历序列构造二叉树

最新推荐文章于 2025-12-31 19:32:08 发布

原创最新推荐文章于 2025-12-31 19:32:08 发布 · 134 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #c++ #数据结构 #leetcode

LeetCode 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种根据二叉树的中序遍历与后序遍历序列来构建二叉树的方法。通过递归地确定根节点，并在中序遍历中找到根节点的位置，从而划分左右子树，最终构建完整的二叉树结构。

从中序与后序遍历序列构造二叉树

题目描述

根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]

返回如下的二叉树：

解题思路

后序遍历是左右根，所以后序数组的最后一个元素就是根节点的值，然后递归，建立子树

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        int pos = inorder.size()-1;
        return buildTree(postorder, pos, inorder, 0, inorder.size() - 1);
    }

    TreeNode* buildTree(vector<int>& postorder, int& pos, vector<int>& inorder, int left, int right) {
        //参数必须是int&pos
        if (pos < 0 || left > right) return 0;
        int p = left;
        while (p <= right && postorder[pos] != inorder[p]) p ++;
        TreeNode* node = new TreeNode(postorder[pos]);
        if (p+1 <= right) node->right = buildTree(postorder, --pos, inorder, p+1, right);
        if (left <= p-1) node->left = buildTree(postorder, --pos, inorder, left, p-1);
        return node;
    }
};