LeetCode 222. Count Complete Tree Nodes

简介: 给出一个完全二叉树,求出该树的节点个数。

v2-af09a872dade52fc0d543ff7972fca67_1440w.jpg

Description



Given a complete binary tree, count the number of nodes.


Note:


Definition of a complete binary tree from Wikipedia:

In a complete binary tree every level, except possibly the last, is completely filled, and all nodes in the last level are as far left as possible. It can have between 1 and 2h nodes inclusive at the last level h.


Example:


Input: 
    1
   / \
  2   3
 / \  /
4  5 6
Output: 6


描述



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


说明:


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


示例:


输入: 
    1
   / \
  2   3
 / \  /
4  5 6

输出: 6


思路



  • 这道题的基本思路是:我们先求左右子树的高度,如果左子树的高度等于右子树的高度,说明左子树是一颗满二叉树,右子树是一刻完全二叉树
  • 如果左子树的高度不等于右子树的高度,说明右子树是一颗满二叉树,左子树不是一颗满二叉树
  • 满二叉树的节点数数为2h-1,再加上当前的根节点,则节点个数为2h,我们在综述中加上2^h,然后我们继续递归的求解完全二叉树的节点个数即可.


# -*- coding: utf-8 -*-
# @Author:             何睿
# @Create Date:        2019-01-27 20:02:19
# @Last Modified by:   何睿
# @Last Modified time: 2019-01-27 20:13:03
# 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):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root: return 0
        # 获取左子树的高度
        left = self.depth(root.left)
        # 获取右子树的高度
        right = self.depth(root.right)
        # 如果左右子树的高度相等,则左子树一定是一颗满二叉树
        if left == right:
            return 2**left + self.countNodes(root.right)
        # 如果左右子树高度不相等,则右子树一定是一颗满二叉树
        if left != right:
            return 2**right + self.countNodes(root.left)
    def depth(self, root):
        # 获取完全二叉树的高度
        if not root: return 0
        res = 0
        while root:
            res += 1
            root = root.left
        return res

源代码文件在这里.


目录
相关文章
|
Java
Leetcode 114. Flatten Binary Tree to Linked List
思路也很简单,先把root的左子树(如有)变成单链表 leftlinkedlist,把root的右子树(如有)变成单链表 rightlinkedlist,再把root的右节点变成leftlikedlist,再把rightlinkedlist接到leftlinkedlist后面,代码如下。
45 1
Leetcode Minimum Depth of Binary Tree (面试题推荐)
计算树的最小深度 很简单的一道题,只需要遍历一次树,到叶子节点的时候计算一下深度和当前最小深度比较,保存最小值就行。 我在这用了一个全局变量 mindepth。总感觉我这代码写的不够简练,求更精简的方法。
47 0
Leetcode Binary Tree Postorder Traversal(面试题推荐)
非递后续归遍历二叉树,肯定得用到栈。先序遍历很好写,但后续遍历就不是那么容易了。 只需要设置个指针pre,指向最后输出的那个节点就行了,只要判断cur指针指向的是上次输出节点的父节点,且cur无其他未遍历的节点,这个时候就把cur节点输出即可,然后更改pre。原理是要遍历当前节点,其所有子节点都必须遍历完,因为肯定是先左后右,所以只需一个指针保持前一次输出的结果即可。
51 0
Leetcode 236. Lowest Common Ancestor of a Binary Tree
根据LCA的定义,二叉树中最小公共祖先就是两个节点p和q最近的共同祖先节点,LCA的定义没什么好解释的,主要是这道题的解法。
38 0
Leetcode 623. Add One Row to Tree
题目很简单,在树的第d层加一层,值为v。递归增加一层就好了。代码如下
47 0
LeetCode contest 190 5418. 二叉树中的伪回文路径 Pseudo-Palindromic Paths in a Binary Tree
LeetCode contest 190 5418. 二叉树中的伪回文路径 Pseudo-Palindromic Paths in a Binary Tree
LeetCode Contest 178-1367. 二叉树中的列表 Linked List in Binary Tree
LeetCode Contest 178-1367. 二叉树中的列表 Linked List in Binary Tree
LeetCode 429. N-ary Tree Level Order Traversal
给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。
86 0
LeetCode 429. N-ary Tree Level Order Traversal
LeetCode 357. Count Numbers with Unique Digits
给定一个非负整数 n,计算各位数字都不同的数字 x 的个数,其中 0 ≤ x < 10n 。
87 0
LeetCode 357. Count Numbers with Unique Digits
|
存储 Python
LeetCode 315. Count of Smaller Numbers After Self
给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。
94 0
LeetCode 315. Count of Smaller Numbers After Self