牛客网：NC84 完全二叉树结点数

原创已于 2022-08-03 12:48:48 修改 · 244 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #c++

于 2022-08-03 12:48:47 首次发布

算法同时被 2 个专栏收录

278 篇文章

订阅专栏

菜鸟刷题记录

204 篇文章

订阅专栏

该博客介绍了两种计算满二叉树节点数的方法：一种是通过递归遍历整个树，另一种利用满二叉树的性质快速计算。详细解析了代码实现过程，包括左右子树高度的获取，以及如何根据高度快速计算节点总数。

直接遍历

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }
}*/
public class Solution {
    public int nodeNum(TreeNode head) {
        if (head == null){
            return 0;
        }
        return 1 + nodeNum(head.left) + nodeNum(head.right);
    }
}

使用满二叉树的性质

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }
}*/
public class Solution {
    public int nodeNum(TreeNode head) {
        if (head == null){
            return 0;
        }
        int left = getHeight(head.left);
        int right = getHeight(head.right);
        if (left != right){
            return (int)Math.pow(2, right) + nodeNum(head.left); 
        } 
        return (int)Math.pow(2, left)  + nodeNum(head.right);
    }
    
    private int getHeight(TreeNode node){
        TreeNode cur = node;
        int res = 0;
        while (cur != null){
            cur = cur.left;
            res++;
        }
        return res;
    }
}