其他分享
首页 > 其他分享> > 【二叉树】leetcode222-完全二叉树的节点个数

【二叉树】leetcode222-完全二叉树的节点个数

作者:互联网

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

完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。
在这里插入图片描述
解答:
方法一:递归

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def countNodes(self, root: TreeNode) -> int:
        if not root:
            return 0
        return self.countNodes(root.left)+self.countNodes(root.right)+1

方法二:迭代

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def countNodes(self, root: TreeNode) -> int:
        if not root:
            return 0
        res=0
        queue=collections.deque()
        queue.append(root)
        while queue:
            #输出队头元素
            t=len(queue)
            res+=t
            for i in range(t):
                node=queue.popleft()
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
        return res

标签:node,leetcode222,self,queue,二叉树,root,节点
来源: https://blog.csdn.net/jqq125/article/details/120972119