刷穿剑指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......

欢迎关注我的公众号: 清风Python,带你每日学习Python算法刷题的同时,了解更多python小知识。

我的个人博客:https://qingfengpython.cn

力扣解题合集:https://github.com/BreezePython/AlgorithmMarkdown

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,444评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,421评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,036评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,363评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,460评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,502评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,511评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,280评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,736评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,014评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,190评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,848评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,531评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,159评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,411评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,067评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,078评论 2 352

推荐阅读更多精彩内容