算法60天-Day16:统计二叉树的深度和结点个数

代码随想录(截图自参考【1】)

本系列是代码随想录算法训练营的学习笔记之day16,主要记录一下刷题的过程,以及核心知识点和一些值的记录的问题。

代码随想录的资源可以看参考链接【1】。

今日知识点

二叉树基础:

  • 二叉树中的几个关键词:结点、度、叶子结点
  • 二叉树的两种主要形式包括满二叉树和完全二叉树
满二叉树(完美二叉树)
  • 满二叉树只有度为0和度为2的结点;
  • 在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2^(h-1) 个节点。
完全二叉树
  • 二叉搜索树:有序树,左子树上所有节点的值小于根结点的值,右子树上所有节点的值大于根结点的值;
二叉搜索树
  • 平衡二叉搜索树:又被称为AVL(Adelson-Velsky and Landis)树,且具有以下性质:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
平衡二叉搜索树
  • 二叉树的储存方式:链式存储
链式存储
顺序存储
  • 用数组存储的二叉树如何遍历呢?如果父节点的下标是i,则左孩子是2i+1,右孩子是2i+2;

  • 二叉树的遍历方式主要有深度优先遍历和广度优先遍历:

    • 深度优先:前序遍历、中序遍历、后序遍历;
    • 广度优先:层次遍
  • 其他的更多理论基础可以去看参考链接【2】;

今日题目

  1. 二叉树的最大深度(104)

给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/maximum-depth-of-binary-tree/

解题思路

  • 递归法理解起来更简单;

  • 要注意根结点的深度为1,其高度就等于整个树的深度;

  • 递归法

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0
        max_l = self.maxDepth(root.left)
        max_r = self.maxDepth(root.right)
        return 1+max(max_l,max_r)
  • 结束递归的条件为root为空,也可以写root==None;
  • 要使用self.maxDepth()来递归调用函数;
  • 迭代法
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
import collections
class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if not root:
            return 0
        depth = 0 #记录深度
        queue = collections.deque()
        queue.append(root)
        while queue:
            size = len(queue)
            depth += 1
            for i in range(size):
                node = queue.popleft()
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
        return depth
  1. 二叉树的最小深度(111)

给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/minimum-depth-of-binary-tree/

解题思路

  • 仍然用递归的方法

  • 递归法

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def minDepth(self, root: Optional[TreeNode]) -> int:
        if root==None:
            return 0
        if not root.left and not root.right:
            return 1
        
        min_d = 1e9
        if root.left:
            min_d = min(self.minDepth(root.left),min_d)
        if root.right:
            min_d = min(self.minDepth(root.right),min_d)
        return min_d+1

  • 结束递归的条件为root为空,也可以写root==None;
  • 结束递归的条件要注意不要忘记了只有当前结点,没有子结点的情况;
  • 迭代法
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def minDepth(self, root: TreeNode) -> int:
        if not root:
            return 0
        que = deque()
        que.append(root)
        res = 1

        while que:
            for _ in range(len(que)):
                node = que.popleft()
                # 当左右孩子都为空的时候,说明是最低点的一层了,退出
                if not node.left and not node.right:
                    return res
                if node.left is not None:
                    que.append(node.left)
                if node.right is not None:
                    que.append(node.right)
            res += 1
        return res
  1. 完全二叉树的结点个数(222)

给你一棵 完全二叉树 的根节点 root ,求出该树的节点个数。

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/count-complete-tree-nodes/

解题思路

  • 首先要清楚的知道完全二叉树的定义,其特点是如果对完美二叉树按照从上至下、从左到右的顺序编号,则完全二叉树中对应结点的序号与完美二叉树中的序号一致;

  • 递归法

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def countNodes(self, root: Optional[TreeNode]) -> int:
        return self.getNodes(root)
    
    def getNodes(self,cur):
        if cur==None:
            return 0
        cnt_l = self.getNodes(cur.left)
        cnt_r = self.getNodes(cur.right)

        return cnt_l+cnt_r+1

  • 只要中间的结点不为None,节点数目就要加上1;
  • getNodes()的参数一定要有self;
  • 下面这种递归更像递归:
class Solution:
    def countNodes(self, root: TreeNode) -> int:
        if not root:
            return 0
        return 1 + self.countNodes(root.left) + self.countNodes(root.right)
  • 迭代法
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
import collections
class Solution:
    def countNodes(self, root: TreeNode) -> int:
        queue = collections.deque()
        if root:
            queue.append(root)
        result = 0
        while queue:
            size = len(queue)
            for i in range(size):
                node = queue.popleft()
                result += 1 #记录节点数量
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
        return result

补充知识点

  1. collections.deque()

collections模块是python内置的模块之一,实现了常见的数据结构,deque是双项队列,具体见参考【2】。

  1. queue.popleft()

移去并且返回一个元素,deque 最左侧的那一个。 如果没有元素的话,就引发 IndexError。

今日思考

  1. 为什么类方法的第一个参数需要为self?

如果不写self的话,就会报错如下:

报错
代码

所以,第一个不写self的话会报错!

class Solution:
    def countNodes(self, root: Optional[TreeNode]) -> int:
        return self.getNodes(root)
    
    def getNodes(self,cur):
        if cur==None:
            return 0
        cnt_l = self.getNodes(cur.left)
        cnt_r = self.getNodes(cur.right)

        return cnt_l+cnt_r+1

实际上self代表的是类对象本身, 比如上面的代码,如果不把self传给getNodes函数,那么在getNodes中就无法调用getNodes函数。。。是有点绕口,但是理解了就好。

来自bing

今日吐槽

百度上搜索一个问题,你们这些垃圾抄过来抄过去,都不看一下答案吗?那答案你们能看懂吗!!就知道cp,就不知道动动脑子!!!

是在下错了,不该用中文搜索。

参考

【1】代码随想录:https://programmercarl.com/
【2】collections:https://docs.python.org/zh-cn/3/library/collections.html?highlight=deque

本文由mdnice多平台发布

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

推荐阅读更多精彩内容