【代码随想录】算法训练计划16

news2024/11/23 15:39:44

【代码随想录】算法训练计划04

1、111. 二叉树的最小深度

题目：
给定一个二叉树，找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明：叶子节点是指没有子节点的节点。

思路：

用递归，后续，高度模拟深度，注意一边为空的情况

func minDepth(root *TreeNode) int {
    // 代码一刷，用递归，后续，高度模拟深度，注意一边为空的情况
    if root == nil {
        return 0
    }
    left := minDepth(root.Left)
    right := minDepth(root.Right)
    if root.Left==nil && root.Right!=nil {
        return 1+right
    }
    if root.Right==nil && root.Left!=nil {
        return 1+left
    }
    return 1+min(left, right)
}
func min(a,b int) int {if a<b {return a}; return b}

2、222. 完全二叉树的节点个数

题目：
给你一棵完全二叉树的根节点 root ，求出该树的节点个数。
完全二叉树的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。

思路：

利用满二叉树特性公式，以及判断深度是否相等计算

func countNodes(root *TreeNode) int {
    //代码一刷
    if root == nil {
        return 0
    }
    left := root.Left
    right := root.Right
    leftDepth, rightDepth := 0, 0
    for left != nil{
        left = left.Left
        leftDepth++
    }
    for right != nil{
        right = right.Right
        rightDepth++
    }
    if leftDepth == rightDepth {
        return (2 << leftDepth) - 1
    }
    return countNodes(root.Left)+countNodes(root.Right)+1
    
}