No.224 - LeetCode[106] Construct Binary Tree from Inorder and Postorder Traversal - 前序后序生成树 - 经典

原创于 2020-09-07 17:45:42 发布 · 129 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种利用后序遍历和中序遍历构建二叉树的方法，核心思想在于后序遍历的最后一个数字为根节点，通过此节点可以划分中序遍历数组，从而确定左右子树的节点数和根节点位置，最终通过深度优先搜索（DFS）完成整个二叉树的构建。

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

核心思想：

后序的最后一个数字是根，划分前序数组
划分同时，左右子数节点数也可以算出，则根的左右儿子可以确定。
最后dfs

/*
 * @lc app=leetcode id=106 lang=cpp
 *
 * [106] Construct Binary Tree from Inorder and Postorder Traversal
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        int N = inorder.size();
        TreeNode* Head = dfs(inorder,0,N,postorder,N-1);
        return Head;
    }
    TreeNode* dfs(vector<int>& inorder,int L,int R,vector<int>& postorder,int loc){
        for(int i=L;i<R;i++){
            if(inorder[i] == postorder[loc]){
                TreeNode* node = new TreeNode(postorder[loc]);
                node->left = dfs(inorder,L,i,postorder,loc-(R-i));
                node->right = dfs(inorder,i+1,R,postorder,loc-1);
                return node;
            }
        }
        return NULL;
    }
};
// @lc code=end