222. Count Complete Tree Nodes
发布日期:2021-05-04 11:02:45 浏览次数:18 分类:技术文章

本文共 1445 字,大约阅读时间需要 4 分钟。

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

来源:力扣(LeetCode)

链接:https://leetcode-cn.com/problems/count-complete-tree-nodes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

完全二叉树的节点数

一、遍历所有节点

二、这是完全二叉树,如果他是一个满二叉树,那么节点数 = 2 ^ 高度 - 1。此时可以判断根节点的左子树或者右子树是不是满二叉树。

判断方法,求出左子树的高度(只用判断跟节点到最左边的节点的高度),如果左子树的高度等于右子树的高度,那说明左子树一定是满二叉树,右子树不一定是满二叉树,那么左子树的节点可以用 2 ^ 高度 - 1计算出,右子树继续遍历。

class Solution {    public int getLeftHight (TreeNode node){        if (node.left == null) {            return 1;        }        return getLeftHight(node.left) + 1;    }    public int countNodes(TreeNode root) {        int ans = 1;        if (root == null) {            return 0;        }        if (root.left == null) {            return 1;        }        if (root.right == null) {            return 2;        }        int leftLen = getLeftHight(root.left);        int rightLend = getLeftHight(root.right);        if (leftLen == rightLend) {            ans += (1 << leftLen) - 1;            ans += countNodes(root.right);        } else {            ans += countNodes(root.left);            ans += countNodes(root.right);        }        return ans;    }}

 

上一篇:229. Majority Element II
下一篇:221. Maximal Square

发表评论

最新留言

不错!
[***.144.177.141]2025年03月23日 13时42分38秒