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

https://leetcode.cn/problems/construct-binary-tree-from-inorder-and-postorder-traversal/

思路:中序和后序遍历分别是左中右和左右中,我们会发现可以先从后序遍历的最后一个节点寻找根节点root,然后从中序遍历找到这个根节点,并从这个节点进行切割,其左右节点分别是左右孩子,然后切割后序数组,这样完成了某层的运算后,进行递归。具体注释详见代码。

代码如下:

class Solution {
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
//我们要返回一个二叉树,所以使用TreeNode*进行函数定义
        if(inorder.size() == 0 || postorder.size() == 0) return NULL;
        return traversal(inorder,postorder);//进行递归
    }
    TreeNode* traversal(vector<int>& inorder, vector<int>& postorder){
//首先设置终止条件,也就是
        if (postorder.size() == 0) return NULL;
//后续数组为0,表示是个空节点,我们返回一个空节点
// 后序遍历数组最后一个元素,就是当前的中间节点,然后定义一个节点,传入数值。以例题为例子,就是定义节点3,但是此时3的左右节点还没定义。
        int rootValue = postorder[postorder.size() - 1];
        TreeNode* root = new TreeNode(rootValue);
//当后续数组大小为1,也就是遍历到了叶子节点,此时返回根节点
        if(postorder.size() == 1) return root;
 // 找到中序遍历的切割点
        int delimiteIndex;
        for(delimiteIndex = 0; delimiteIndex < inorder.size();delimiteIndex++){
            if(inorder[delimiteIndex] == rootValue) break;
        }
 // 切割中序数组
        vector<int> leftInorder(inorder.begin() , inorder.begin() + delimiteIndex);
        vector<int> rightInorder(inorder.begin() + delimiteIndex + 1, inorder.end());
        postorder.resize(postorder.size() - 1);// postorder 舍弃末尾元素
// 切割后序数组
        vector<int> leftPostorder(postorder.begin(), postorder.begin() + leftInorder.size());
        vector<int> rightPostorder( postorder.begin() + leftInorder.size(),postorder.end());

        
        root -> left = traversal(leftInorder,leftPostorder);
        root -> right = traversal(rightInorder,rightPostorder);
        return root;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值