[LeetCode] 106. Construct Binary Tree from Inorder and Postorder Traversal

45 篇文章 0 订阅

原题链接:https://leetcode.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal/

1. 题目介绍

Given inorder and postorder traversal of a tree, construct the binary tree.
给出一个树的中序遍历和后序遍历顺序,构造一棵二叉树

Note:
You may assume that duplicates do not exist in the tree.
你可以假设,树中没有重复的节点

For example, given
比如,分别给出中序遍历和后序遍历的路径:
inorder = [9,3,15,20,7]
postorder = [9,15,7,20,3]

Return the following binary tree:
需要返回下面这样的二叉树:
在这里插入图片描述

类似题目

105. Construct Binary Tree from Preorder and Inorder Traversal
给出一棵树的前序遍历和中序遍历,根据这两种遍历顺序构造一棵二叉树。

2. 解题思路

方法1 递归

  1. 在后序遍历中找到根节点
    后序遍历的最后一个数必定是根节点,因此我们可以轻松知道根节点是谁。

  2. 在中序遍历中找到根节点
    由题目可知,树中没有重复的节点,于是我们可以在中序遍历中找到根节点。

  3. 在中序遍历中找到左右子树的区间
    中序遍历中,根节点之前的部分就是左子树,根节点之后的树就是右子树。同时可以计算出左子树的节点个数。

  4. 在后序遍历中找到左右子树的区间
    根据左子树的长度,我们回到后序遍历,又可以找出在后序遍历中,左子树和右子树的区间在哪里。

  5. 递归重复上述4步,每一步都返回当前的根节点和左右子树。左右子树交给下一层递归完成计算。

实现代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        return helper(inorder,0, inorder.length-1 , postorder, 0, postorder.length-1);
    }
    public TreeNode helper(int[] inorder , int in_start, int in_end, int[] postorder, int po_start, int po_end ){
        if(in_start > in_end || po_start > po_end){
            return null;
        }
        int rootVal = postorder[po_end];//后序遍历的最后一个值必定是根节点
        
        //根据根节点的值,在中序遍历顺序中找到根节点的下标,存放于rootIndex
        int rootIndex = 0;
        for(int i = in_end; i >= in_start ; i--){
            if(inorder[i] == rootVal){
                rootIndex = i;
                break;
            }
        }
        
        //在中序遍历数组中,计算左子树的节点个数
        int length = rootIndex - in_start;
        
        TreeNode root = new TreeNode(rootVal);
        root.left = helper(inorder, in_start, rootIndex-1, postorder, po_start, po_start+length-1);
        //左子树的区间,在中序遍历的数组中,是[in_start, rootIndex-1],
        //在后序遍历的数组中,是[po_start, po_start+length-1]
        
        root.right= helper(inorder, rootIndex+1, in_end, postorder, po_start+length,po_end-1);
        //右子树的区间,在中序遍历的数组中,是[rootIndex+1, in_end],
        //在后序遍历的数组中,是[po_start+length, po_end-1]
        
        return root;
    }
}

方法2 递归+HashMap

是对方法1的改进,使用HashMap存放中序遍历的数值和下标,当需要求根节点的下标时就可以利用hashMap找出来。
实现方法

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int [] in ;
    int [] po ;
    HashMap <Integer, Integer> map = new HashMap<>();
    int po_rootIndex ;
    
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        in = inorder;
        po = postorder;
        po_rootIndex = postorder.length - 1;
        
        int Count = 0;
        for(int i : inorder){
            map.put(i , Count);
            Count++;
        }
        
        return helper(0, in.length-1);
    }
    public TreeNode helper(int l , int r){
        if(l > r){
            return null;
        }
        
        int rootVal = po[po_rootIndex];//在后序遍历中找到根节点的值
        po_rootIndex --;
        
        int in_rootIndex = map.get(rootVal);//在中序遍历中找到根节点的下标
        
        TreeNode root = new TreeNode(rootVal);
        
        //此处必须先构建右子树,再构建左子树
        //因为后序遍历的数组,根节点的下标向前是右子树,再向前才是左子树
        root.right = helper(in_rootIndex+1 , r);
        root.left  = helper(l , in_rootIndex-1);
        
        return root;
    }
}

3. 参考资料

https://blog.csdn.net/Bertram03/article/details/92856130

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值