代码随想录打卡 Day16

最新推荐文章于 2024-07-18 10:21:10 发布

不见星河路

最新推荐文章于 2024-07-18 10:21:10 发布

阅读量107

点赞数

文章标签： c++ 算法开发语言

本文链接：https://blog.csdn.net/qq_43593188/article/details/129374136

版权

本文提供了几种树结构（二叉树、n叉树）的最大深度、最小深度以及完全二叉树节点个数的计算方法。通过递归和广度优先搜索（BFS）策略，实现了对树的深度遍历并计算相关属性。

摘要由CSDN通过智能技术生成

104.二叉树的最大深度

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func maxDepth(root *TreeNode) int {
    var seekDepth func(root *TreeNode,depth int)
    if root == nil{
        return 0
    }
    res := 0
    //queue := list.New()
    seekDepth = func(root *TreeNode,depth int){
        if root == nil{
            return 
        }
        if(res < depth){
            res = depth
        }
        seekDepth(root.Left,depth+1)
        seekDepth(root.Right,depth+1)
    }
    seekDepth(root,0)
    return res + 1
}

559.n叉树的最大深度

/**
 * Definition for a Node.
 * type Node struct {
 *     Val int
 *     Children []*Node
 * }
 */

func maxDepth(root *Node) int {
    count := 0
    if root == nil {
        return count
    }
    ss := list.New()
    ss.PushBack(root)
    for ss.Len() > 0{
        length := ss.Len()
        count++
        for i := 0 ; i < length ; i++{
            node := ss.Remove(ss.Front()).(*Node)
            for j := 0 ; j < len(node.Children) ; j++{
                ss.PushBack(node.Children[j])
            }
        }
    }
    return count
}

● 111.二叉树的最小深度

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func minDepth(root *TreeNode) int {
    queue := list.New()
    if root == nil{
        return 0
    }
    queue.PushBack(root)
    depth := 0
    for queue.Len() > 0{
        length := queue.Len()
        depth = depth + 1
        for i := 0 ; i < length ; i++{
            node := queue.Remove(queue.Front()).(*TreeNode)
            if(node.Left == nil && node.Right == nil){
                return depth
            }
            if(node.Left != nil){
                queue.PushBack(node.Left)
            }
            if(node.Right != nil){
                queue.PushBack(node.Right)
            }
        }
    }

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

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func countNodes(root *TreeNode) int {
    ss := list.New()
    count := 0
    if root == nil {
        return count
    }
    ss.PushBack(root)
    for ss.Len() > 0{
        length := ss.Len()
        for i := 0 ; i < length ; i++{
            node := ss.Remove(ss.Front()).(*TreeNode)
            count++
            if(node.Left != nil){
                ss.PushBack(node.Left)
            }
            if(node.Right != nil){
                ss.PushBack(node.Right)
            }
        }
    }
    return count
}