您现在的位置是：首页 >技术杂谈 >leetcode刷题（9）二叉树（3）网站首页技术杂谈

leetcode刷题（9）二叉树（3）

不能再留遗憾了 2023-06-16 16:00:02
简介leetcode刷题（9）二叉树（3）

各位朋友们，提前祝大家五一劳动节快乐啊！！！今天我为大家分享的是关于leetcode刷题二叉树相关的第三篇我文章，让我们一起来看看吧。

文章目录

1.二叉树的层序遍历

leetcode之二叉树的层序遍历（难度：中等）

题目要求

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

示例1
在这里插入图片描述
输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]

示例2
输入：root = [1]
输出：[[1]]

示例3
输入：root = []
输出：[]

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {

    }
}

做题思路

这道题我们可以借助队列这个数据结构，首先先将根节点放进队列中，然后我们计算此时队列的大小，记录下来，这个数字其实就是各层根节点的个数。我们等会要弹出相同数量的元素，在弹出的同时我们需要将弹出的根节点放进List中，并且将次根节点的非空左树和右树插入队列中。我们举个例子。
在这里插入图片描述

在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
最后我们返回List就是我们需要的结果。

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> list = new ArrayList<>();
        //当根节点为null时，我们直接返回list，这里list也为null
        if(root == null) {
            return list;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()) {
            int size = queue.size();
            List<Integer> tmp = new ArrayList<>();
            while(size > 0) {
                TreeNode cur = queue.poll();
                tmp.add(cur.val);
                if(cur.left != null) {
                    queue.offer(cur.left);
                }
                if(cur.right != null) {
                    queue.offer(cur.right);
                }
                size--;
            }
            list.add(tmp);
        }
        return list;
    }
}

在这里插入图片描述

2.从前序与中序遍历序列构造二叉树

leetcode之从前序与中序遍历序列构造二叉树（难度：中等）

题目要求

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

示例 1:
在这里插入图片描述

输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

示例 2:

输入: preorder = [-1], inorder = [-1]
输出: [-1]

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {

    }
}

做题思路

我们通过二叉树的前序遍历序列来确定根节点，通过中序遍历来确定左右子树，我们直接来举一个例子。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

用inBegin和inEnd来分别表示左右子树的范围，当表示左树时inEnd = index-1
表示右树时inBegin = index+1
在这里插入图片描述
此时9的左树为null，说明3的左树的树叶是9，也就是说我们递归结束的条件是inBegin > inEnd
在这里插入图片描述

在这里插入图片描述

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
//创建一个成员变量用来遍历前序遍历序列
    public int i = 0;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
    //我们inBegin默认为0，inEnd默认数组的最后一个元素的下标
        return buildTreeChild(preorder,inorder,0,inorder.length-1);
    }

    public TreeNode buildTreeChild(int[] preorder,int[] inorder,int inBegin, int inEnd) {
    //递归结束的条件
        if(inBegin > inEnd) {
            return null;
        }
        TreeNode root = new TreeNode(preorder[i]);
        int index = findIndex(preorder[i],inorder,inBegin,inEnd);
        i++;
        root.left = buildTreeChild(preorder,inorder,inBegin,index-1);
        root.right = buildTreeChild(preorder,inorder,index+1,inEnd);

        return root;
    }

    public int findIndex(int key,int[] inorder,int inBegin,int inEnd) {
        for(int j = inBegin; j<= inEnd; j++) {
            if(key == inorder[j]) {
                return j;
            }
        }

        return -1;
    }
}

在这里插入图片描述

3.从中序与后序遍历序列构造二叉树

leetcode之从中序与后序遍历序列构造二叉树（难度：中等）

题目要求

给定两个整数数组 inorder 和 postorder ，其中 inorder 是二叉树的中序遍历， postorder 是同一棵树的后序遍历，请你构造并返回这颗 二叉树 。

示例 1:
在这里插入图片描述

输入：inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
输出：[3,9,20,null,null,15,7]

示例 2:

输入：inorder = [-1], postorder = [-1]
输出：[-1]

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {

    }
}

做题思路

这道题的思路其实和上面的那个题差不多的，唯一不同的是我们需要从后序遍历序列的后面开始遍历，并且上一个题是先构建左子树，而我们这个题需要先构建右子树，因为后序遍历的顺序是先左子树、再右子树、最后是根节点，我们是从后序遍历的最后面开始的，所以我们构建的顺序也是不一样的。我们就直接来看代码吧。

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public int i = 0;
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        i = postorder.length-1;
        return buildTreeChild(inorder,postorder,0,postorder.length-1);
    }

    public TreeNode buildTreeChild(int[] inorder,int[] postorder,int inBegin,int inEnd) {
        if(inBegin > inEnd) {
            return null;
        }
        TreeNode root = new TreeNode(postorder[i]);
        int index = findIndex(postorder[i],inorder,inBegin,inEnd);
        i--;
        root.right = buildTreeChild(inorder,postorder,index+1,inEnd);
        root.left = buildTreeChild(inorder,postorder,inBegin,index-1);

        return root;
    }

    public int findIndex(int key, int[] inorder, int inBegin, int inEnd) {
        for(int j = inBegin; j <= inEnd; j++) {
            if(inorder[j] == key) {
                return j;
            }
        }
        return -1;
    }
}

在这里插入图片描述

风语者！平时喜欢研究各种技术，目前在从事后端开发工作，热爱生活、热爱工作。

站长推荐