LeetCode每日1题--从中序与后序遍历序列构造二叉树

56 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第5天,点击查看活动详情

前言

算法的重要性不言而喻!区分度高!

现在学习的门槛低了,只有能上网每个人都可以学编程!培训班6个月就可以培养出来能干活的人,你怎么从这些人中脱颖而出?没错!就是学算法,学一些底层和基础的东西。

说的功利点是为了竞争,卷死对手。真心话说就是能提高自己的基础能力,为技术可持续发展做好充分的准备!!!

提前入门学习书籍:CPrimerPlus、大话数据结构

image-20220705103735001

刷题网站

代码随想录 (programmercarl.com)

leetcode

我是按照代码随想录提供的刷题顺序进行刷题的,大家也可以去刷leetcode最热200道,都可以

刷题嘛,最重要的就是坚持了!!!

画图软件

OneNote

这个要经常用,遇见不懂的流程的话就拿它画一画!

笔记软件

Typoral

题目

image.png

解析

递归三部曲

  1. 确定递归的参数和返回值 返回值最后要构造好的二叉树,所以返回TreeNode

参数:

  1. 提供的中序遍历数组和起始,结束点
  2. 提供的后序遍历数组和起始,结束点
public TreeNode findNode(int[] inorder, int inBegin, int inEnd, int[] postorder, int postBegin, int postEnd) {}
  1. 确定递归终止的条件

参数里的范围都是前闭后开,这里为边界条件的判断

// 不满足左闭右开,说明没有元素,返回空树
if (inBegin >= inEnd || postBegin >= postEnd) {  
    return null;
}
  1. 确定单层递归的逻辑

这里其实是固定的写法,看上去很多其实逻辑很简单的。就是我们思路的实现

先通过int rootIndex = map.get(postorder[postEnd - 1]);找到后序遍历的最后一个元素在中序遍历中的位置

之后构造节点 TreeNode root = new TreeNode(inorder[rootIndex]);

接着 int lenOfLeft = rootIndex - inBegin; 计算分割后的中序左子树个数,用来确定后序数列的个数

最后就是 递归构造左子树和右子树

int rootIndex = map.get(postorder[postEnd - 1]);   
TreeNode root = new TreeNode(inorder[rootIndex]); 
int lenOfLeft = rootIndex - inBegin;  

root.left = findNode(inorder, inBegin, rootIndex,
                    postorder, postBegin, postBegin + lenOfLeft);
                    
root.right = findNode(inorder, rootIndex + 1, inEnd,
                    postorder, postBegin + lenOfLeft, postEnd - 1);

一图胜千言

这里主要要掌握的点就是知道中序和后序遍历如何确定二叉树

确定的方法就是用后续数组的最后一个元素为切割点,先切割中序数组,根据中序数组,反过来切割后序数组。

自己不懂的话可以画图尝试理解。

image.png

完整代码


class Solution {
    Map<Integer, Integer> map;  // 方便根据数值查找位置
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        map = new HashMap<>();
        for (int i = 0; i < inorder.length; i++) { // 用map保存中序序列的数值对应位置
            map.put(inorder[i], i);
        }

        return findNode(inorder,  0, inorder.length, postorder,0, postorder.length);  // 前闭后开
    }
    
    public TreeNode findNode(int[] inorder, int inBegin, int inEnd, int[] postorder, int postBegin, int postEnd) {
        if (inBegin >= inEnd || postBegin >= postEnd) {  
            return null;
        }
        int rootIndex = map.get(postorder[postEnd - 1]); 
        TreeNode root = new TreeNode(inorder[rootIndex]);  
        int lenOfLeft = rootIndex - inBegin;  
        root.left = findNode(inorder, inBegin, rootIndex,
                            postorder, postBegin, postBegin + lenOfLeft);
        root.right = findNode(inorder, rootIndex + 1, inEnd,
                            postorder, postBegin + lenOfLeft, postEnd - 1);

        return root;
    }
}