06-03.统计完全二叉树的节点个数

题目

给定一棵完全二叉树的头节点head,返回这棵树的节点个数.

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int countNodes(TreeNode root) {
        if(root == null){
            return 0;
        }
        return bs(root, 1, mostLeftLevel(root, 1));
    }

    private int mostLeftLevel(TreeNode node, int level){
        while(node != null){
            node = node.left;
            level ++;
        }
        return level - 1;
    }

    private int bs(TreeNode node, int level, int h){
        if(level == h){
            return 1;
        }
        if(mostLeftLevel(node.right, level + 1) == h){
            return (1 << (h - level)) + bs(node.right, level + 1, h);
        }else{
            return (1 << (h - level - 1)) + bs(node.left, level + 1, h);
        }
    }
}
©️2020 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值