Files
leetcode-master/添加0222.完全二叉树的节点个数Go版本.md
Galaxies2580 f59621fdc5 添加0222.完全二叉树的节点个数Go版本
添加0222.完全二叉树的节点个数Go版本
2022-08-15 10:24:40 +08:00

501 B

func countNodes(root *TreeNode) int {
    if root == nil {
        return 0
    }
    q := list.New()
    q.PushBack(root)
    res := 0
    for q.Len() > 0 {
        n := q.Len()
        for i := 0; i < n; i++ {
            node := q.Remove(q.Front()).(*TreeNode)
            if node.Left != nil {
                q.PushBack(node.Left)
            }
            if node.Right != nil {
                q.PushBack(node.Right)
            }
            res++
        }
    }
    return res 
}