刷穿剑指offer-Day21-队列II 使用队列实现广度优先搜索!

简介: 刷穿剑指offer-Day21-队列II 使用队列实现广度优先搜索!

昨日回顾


昨天,我们介绍了队列的基础知识,并分别学习了Python与Java中队列的操作方法。

在此基础上,通过两道简单的队列题,掌握了基本的入队与出队操作。然而,队列的题目可不只是简单按照顺序出入队那么简单。

队列有一个最为经典的使用方式,就是 广度优先搜索( BFS) 。而广度优先搜索中,又数二叉树的遍历更为常用。今天我们就通过几道二叉树的题目,来掌握广度优先搜索的使用。

二叉树什么时候该使用广度优先搜索呢?请大家注意这几个关键字: 每层、最大深度、最左、最右


二叉树BFS模板


虽然昨天我们学习的队列题目,看似与二叉树没有关系,但其实二叉树的广度优先搜索,是存在解题模板可以套用的。

对于基本的二叉树层序遍历,使用如下的模板基本就可以解决所有问题了:

初始化队列 queue
if root不为空:
    将root加入队列
while queue 不为空:
    length = queue的长度
    for (int i = 0; i < lg; i++)
        q = queue队首出队
        获取q.val
        # 执行题目相关操作
        分别判断 q.left、q.right是否存在:
           将q.left、q.right 加入队列
return 最终结果

除了模板中的相关操作,其他框架基本都是固定的,下面来看一道基础的广度优先层序遍历问题吧!


剑指OfferII044.二叉树每层的最大值


https://leetcode-cn.com/problems/hPov7L/solution/shua-chuan-jian-zhi-offer-day21-dui-lie-vo9a5/

难度:中等


题目

给定一棵二叉树的根节点 root ,请找出该二叉树中每一层的最大值。

提示:

  • 二叉树的节点个数的范围是 [0,10 ^ 4]
  • -2 ^ 31 <= Node.val <= 2 ^ 31 - 1


示例

示例1:
输入: root = [1,3,2,5,3,null,9]
输出: [1,3,9]
解释:
          1
         / \
        3   2
       / \   \  
      5   3   9 
示例2:
输入: root = [1,2,3]
输出: [1,3]
解释:
          1
         / \
        2   3
示例3:
输入: root = [1]
输出: [1]
示例4:
输入: root = [1,null,2]
输出: [1,2]
解释:      
           1 
            \
             2     
示例5:
输入: root = []
输出: []


分析

题目要求我们获取每一层的最大值,正好满足队列的出入队操作。

  1. 首先我们需要初始化队列queue、及返回队列ret
  2. 当队列不为空时,循环执行出入队
  3. 访问每一层前,初始化num为最小值
  4. 然后比较获取当前队列中的val最大值,并将最大值加入ret中
  5. 根据每个出队的节点,判断该节点的左、右子树是否存在,若存在则执行入队操作
  6. 循环3--5步骤,直到队列为空停止,并返回ret即可


解题


Python:

class Solution:
    def largestValues(self, root):
        ret, queue = [], deque()
        if root:
            queue.append(root)
        while queue:
            num = -float('inf')
            for i in range(len(queue)):
                q = queue.popleft()
                num = max(num, q.val)
                if q.left:
                    queue.append(q.left)
                if q.right:
                    queue.append(q.right)
            ret.append(num)
        return ret


Java:

class Solution {
    public List<Integer> largestValues(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        List<Integer> ret = new LinkedList<>();
        if (root != null) {
            queue.add(root);
        }
        while (!queue.isEmpty()) {
            int num = Integer.MIN_VALUE;
            int lg = queue.size();
            for (int i = 0; i < lg; i++) {
                TreeNode q = queue.poll();
                num = Math.max(num, q.val);
                if (q.left != null) {
                    queue.add(q.left);
                }
                if (q.right != null) {
                    queue.add(q.right);
                }
            }
            ret.add(num);
        }
        return ret;
    }
}

通过上面的题目,我们可以发现,只是条件判断略有不同,而代码总体没有太大的差别。不信,再来看看下面的题目吧。


剑指OfferII045.二叉树最底层最左边的值


https://leetcode-cn.com/problems/LwUNpT/solution/shua-chuan-jian-zhi-offer-day21-dui-lie-do26g/

难度:中等


题目

给定一个二叉树的 根节点 root,请找出该二叉树的 最底层 最左边 节点的值。

假设二叉树中至少有一个节点。

提示:

  • 二叉树的节点个数的范围是 [1,10 ^ 4]
  • -2 ^ 31 <= Node.val <= 2 ^ 31 - 1


示例:

示例 1:
输入:
    2
   / \
  1   3
输出:
1
示例 2:
输入:
        1
       / \
      2   3
     /   / \
    4   5   6
       /
      7
输出:
7


分析

首先,最简单的应该是BFS逐层遍历

  1. 我们创建一个变量ret,用于记录每行的第一个val
  2. 然后创建队列queue,由于题目说明至少有一个节点,则root无脑入队,开始while循环
  3. 判断每行的第一个节点,将ret变量更新为首个节点的值
  4. 直到队列为空时,终止循环,返回ret即可。


解题


Python:

class Solution:
    def findBottomLeftValue(self, root):
        queue = deque([root])
        ret = root.val
        while queue:
            for i in range(len(queue)):
                q = queue.popleft()
                if i == 0:
                    ret = q.val
                if q.left:
                    queue.append(q.left)
                if q.right:
                    queue.append(q.right)
        return ret


Java:

class Solution {
    public int findBottomLeftValue(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int ret = root.val;
        while (!queue.isEmpty()) {
            int lg = queue.size();
            for (int i = 0; i < lg; i++) {
                TreeNode q = queue.poll();
                if (i == 0){
                    ret = q.val;
                }
                if (q.left != null) {
                    queue.add(q.left);
                }
                if (q.right != null) {
                    queue.add(q.right);
                }
            }
        }
        return ret;
    }
}

最后再来看一道类似的题目吧,通过这三道二叉树的广度优先搜索题目,相信大家对于这方面的知识点,可以完全掌握了。


剑指OfferII046.二叉树的右侧视图


https://leetcode-cn.com/problems/WNC0Lk/solution/shua-chuan-jian-zhi-offer-day21-dui-lie-n360i/

难度:中等


题目

给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。

提示:

  • 二叉树的节点个数的范围是 [0,100]
  • -100 <= Node.val <= 100


示例

示例 1:
        1
       / \
      2   3
       \   \
        5   4
输入: [1,2,3,null,5,null,4]
输出: [1,3,4]
示例 2:
输入: [1,null,3]
输出: [1,3]
示例 3:
输入: []
输出: []


分析

这道题,使用简单的BFS逐层遍历即可解题

  1. 我们创建一个列表ret,用于记录每行的最右节点的val
  2. 然后创建队列queue,当root不为空时,将root加入队列,然后开始while循环
  3. 判断每行的最后一个节点,并将该节点的值加入ret列表中
  4. 根据每个出队的节点,判断该节点的左、右子树是否存在,若存在则执行入队操作
  5. 直到队列为空时,终止循环,返回ret即可


解题


Python:

class Solution:
    def rightSideView(self, root):
        queue, ret = deque(), []
        if root:
            queue.append(root)
        while queue:
            lg = len(queue)
            for i in range(lg):
                q = queue.popleft()
                if i == lg - 1:
                    ret.append(q.val)
                if q.left:
                    queue.append(q.left)
                if q.right:
                    queue.append(q.right)
        return ret


Java:

class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        if (root != null) {
            queue.add(root);
        }
        List<Integer> ret = new LinkedList<>();
        while (!queue.isEmpty()) {
            int lg = queue.size();
            for (int i = 0; i < lg; i++) {
                TreeNode q = queue.poll();
                if (i == lg - 1) {
                    ret.add(q.val);
                }
                if (q.left != null) {
                    queue.add(q.left);
                }
                if (q.right != null) {
                    queue.add(q.right);
                }
            }
        }
        return ret;
    }
}

关于队列的题目,就介绍到这里,当然通过二叉树的广度优先搜索,也引入了树这个概念。明天,我们就该进入树这章的专题了!


to be continue......



相关文章
|
8月前
【一刷《剑指Offer》】面试题 7:用两个栈实现队列
【一刷《剑指Offer》】面试题 7:用两个栈实现队列
|
8月前
|
C语言
Leetcode每日一题——“用栈实现队列”
Leetcode每日一题——“用栈实现队列”
|
8月前
|
Java C++ Python
leetcode-232:用栈实现队列
leetcode-232:用栈实现队列
51 0
|
8月前
|
Java C++
【剑指offer】-两个栈来实现一个队列-05/67
【剑指offer】-两个栈来实现一个队列-05/67
|
C语言
[链表OJ题 8] 用栈实现队列,没想到你小子的基础这么好,这么快就做对了
[链表OJ题 8] 用栈实现队列,没想到你小子的基础这么好,这么快就做对了
|
Python
LeetCode 232. 用栈实现队列
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty
69 0
|
算法 Java
算法打卡Day15_leetcode _232 用栈实现队列
算法打卡Day15_leetcode _232 用栈实现队列
算法打卡Day15_leetcode _232 用栈实现队列
|
算法 Java C++
用栈实现队列(LeetCode 232)
用栈实现队列(LeetCode 232)
124 0
C++解LeetCode232. 用栈实现队列(适合基础薄弱)
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty): 实现 MyQueue 类: void push(int x) 将元素 x 推到队列的末尾 int pop() 从队列的开头移除并返回元素 int peek() 返回队列开头的元素 boolean empty() 如果队列为空,返回 true ;否则,返回 false
91 0
C++解LeetCode232. 用栈实现队列(适合基础薄弱)
leetcode 用栈实现队列
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty): 实现 MyQueue 类: void push(int x) 将元素 x 推到队列的末尾 int pop() 从队列的开头移除并返回元素 int peek() 返回队列开头的元素 boolean empty() 如果队列为空,返回 true ;否则,返回 false
72 0