记录*:LeetCode 222

这道题我没做出来,十分挫败。看了答案之后,不仅被作者精妙的思路折服,并且还学习到了一些小技巧。
这里直接贴上作者原文

题目描述

在这里插入图片描述

错误思路

在这里插入图片描述
上面是我的错误思路。我的思路是,通过最左边的路径找到树的高度。然后寻求所有高度为 h-1的节点,最后再计算缺失的节点数,即可得到存在的总数。
然而,事实时,当我在遍历高度为 h-1 的节点时,我始终找不到一个好的办法,来进行横向扫描,我总不能进行层序遍历吧,虽然可以解出来,但是把整个树遍历一遍,题目的:完全二叉树这一条件就会完全没有用。
我甚至尝试用一个栈来维护所有经过的父节点,然后通过栈的长度来推测是否目前在 h-1 高度。但是在涉及到跨树时,频繁的栈操作导致栈会十分难以维护。总之,写不下去了。

正确的解法

直接引用作者原话:

首先需要明确完全二叉树的定义:它是一棵空树或者它的叶子节点只出在最后两层,若最后一层不满则叶子节点只在最左侧。
再来回顾一下满二叉的节点个数怎么计算,如果满二叉树的层数为h,则总节点数为:2^h - 1.
那么我们来对 root 节点的左右子树进行高度统计,分别记为 left 和 right,有以下两种结果:

  1. left == right。这说明,左子树一定是满二叉树,因为节点已经填充到右子树了,左子树必定已经填满了。所以左子树的节点总数我们可以直接得到,是 2 l e f t − 1 2^{left - 1} 2left1,加上当前这个 root 节点,则正好是 2 l e f t 2^{left} 2left。再对右子树进行递归统计。
  2. left != right。说明此时最后一层不满,但倒数第二层已经满了,可以直接得到右子树的节点个数。同理,右子树节点 + root 节点,总数为 2 r i g h t 2^{right} 2right。再对左子树进行递归查找。

作者:xiao-ping-zi-5
链接:https://leetcode-cn.com/problems/count-complete-tree-nodes/solution/chang-gui-jie-fa-he-ji-bai-100de-javajie-fa-by-xia/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

作者从高度出发,利用递归,解决了这个问题。
反思我的思考过程,一直把眼光盯在了如何操作最后一层,殊不知,高度本身就可以得出节点的个数,利用递归不断将高度变小,问题规模缩小,那么最后就能够得到节点个数。

那么贴上作者的代码,代码也是十分精炼优秀。

/**
 * 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;
        } 
        int left = countLevel(root.left);
        int right = countLevel(root.right);
        if(left == right){
            return countNodes(root.right) + (1<<left);
        }else{
            return countNodes(root.left) + (1<<right);
        }
    }
    private int countLevel(TreeNode root){
        int level = 0;
        while(root != null){
            level++;
            root = root.left;
        }
        return level;
    }
}

作者:xiao-ping-zi-5
链接:https://leetcode-cn.com/problems/count-complete-tree-nodes/solution/chang-gui-jie-fa-he-ji-bai-100de-javajie-fa-by-xia/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

从上述代码也可以学到,假如想要计算 2 n 2^n 2n,直接通过移位左移即可。同理,对于 2 − n 2^{-n} 2n,则为右移。1 << left 即为 2 l e f t 2^{left} 2left

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值