从中序和后序遍历构造二叉树(递归)

最新推荐文章于 2024-07-18 10:27:47 发布

Stdleohao

最新推荐文章于 2024-07-18 10:27:47 发布

阅读量390

点赞数

分类专栏： nowcode 文章标签： dfs 数据结构二叉树算法

本文链接：https://blog.csdn.net/stdleohao/article/details/115802321

版权

nowcode 专栏收录该内容

38 篇文章 0 订阅

订阅专栏

题目描述

给出一棵树的中序遍历和后序遍历，请构造这颗二叉树注意：保证给出的树中不存在重复的节点

思路

思路很容易想，但是能不能顺利写出代码来又是另外一回事。
我使用了开区间
后序遍历的左右子树的区分应该由中序遍历得到的左子树长度来固定，而不是由中序遍历左子树的下标。

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param inorder int整型vector 
     * @param postorder int整型vector 
     * @return TreeNode类
     */
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        // write code here
        // 中序遍历：左中右
        // 后序遍历：左右中
        int i = inorder.size();
        int p = postorder.size();
        if(p == 0){
            return NULL;
        }
        return compare(inorder, 0, i,  postorder, 0, p);
    }
    
    TreeNode* compare(vector<int>& inorder, int Il, int IR, vector<int>& postorder, int Pl, int PR){
        int mid ;
        if(Il>=IR)
            return NULL;
        TreeNode* root = new TreeNode(postorder[PR-1]);
        for(int i=Il; i<IR; i++){
            if(inorder[i] == postorder[PR-1]){
                mid = i;
                break;
            }
        }
        int leftlen = mid - Il;
        root->left = compare(inorder, Il, mid,  postorder, Pl, Pl+leftlen);
        root->right = compare(inorder, mid+1, IR,  postorder, Pl+leftlen, PR-1);
        return root;
    }
    
    
};