算法学习leetcode 222题完全二叉树的节点个数

给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。

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

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

 思考：根据题目我们可以知道，这个树由一个根节点，每个根节点有最多两个子节点，如果只有一个子节点，那么它最多只有一个子左节点，并且如果一层没满的话，并且所有节点是从左往分面布的，如果第n-1层没有满，则n层不会有节点。

//定义方法为当前包含当前节点,以及此节点往下面所有节点的数量.所以如果当前结果为null,则返回0 不会进入方法
//如果当前节点只有左节点,则返回 1+countNodes(左节点),如情况B;如果当前左右子节点都有,
// 则返回1+countNodes(root.left) + countNodes(root.right),如情况C;
//如果左右子结点都无,则返回1,如情况D

public int countNodes(TreeNode root) {
    if (root == null) {
        return 0;
    }
    if (root.right != null) {
        return 1 + countNodes(root.left) + countNodes(root.right);
    } else if (root.left != null) {
        return 1 + countNodes(root.left);
    }
    return 1;
}

//钉2023/0302：这里注意前面已经对null做判断了，后面再用两个if判断就有点多余了，可以改为

public int countNodes(TreeNode root) {
    if (root == null) {
        return 0;
    }
        return 1 + countNodes(root.left) + countNodes(root.right);
}

进阶：遍历树来统计节点是一种时间复杂度为 O(n) 的简单解决方案。你可以设计一个更快的算法吗？

思考：观察发现节点上面的val都是按顺序来的，并且n层的节点数为  2^n-1   ,从根节点到每个节点的路径是不一样的，比如 4 5 6 7的路径分别为  左左  左右  右左  右右，相信大家也发现了，就是 n-1个 左右字符组成的长度为n-1的字符串按照字典序（左小于右）排序。所以问题就变成了求最大深度对应的 左右字符的组成的字符串的数组。

public void dfs(List<String> list, int a,String s){
//左右分别用 'A' 'B'表示
    if(s.length()==a){
        list.add(s);
        return;
    }
    dfs(list,a,s+'A');
    dfs(list,a,s+'B');
}

这个数组也可以用integer转二进制来获得

public List<String>  dfs2(List<String> list,int a){
    for (int i = 0; i < aPowB(2, a); i++) {
        list.add(getNs(Integer.toBinaryString(i),a));
    }
    return list;
}
public String getNs(String s,int n){
    if(n!=s.length()){
        for (int i = 0; n != s.length(); i++) {
            s="0"+s;
        }
    }
    return s.replace("0","A").replace("1","B");
}

下面求最大深度：

public int getLongr(TreeNode root){
    if(root.left!=null){
        return 1+getLongr(root);
    }
    return 1;
}

再写个根据指令（字符串)获取当前节点的方法（node or null）

public TreeNode getNodeVal(TreeNode root, String s) {
    if (s.length() == 0) {
        return root;
    }
    return s.charAt(0) == 'A' ? getNodeVal(root.left, s.substring(1)) : getNodeVal(root.right, s.substring(1));
}

 最后我们可以用二分法进行定位node

public int getAns(int a,int b,List<String> list,TreeNode root){
    if(getNodeVal(root,list.get((a+b)/2))==null){
        if(a==b)
            return a-1;
        return getAns(a,Math.min(b-1,(a+b)/2),list,root);
    }
    if(a==b){
        return a;
    }
    return getAns(Math.max(a+1,(a+b)/2),b,list,root);
}

执行一下速度慢的要死

 考虑了一下，用dfs计算指令集的时候，计算了太多的无用数据，这些数据的计算量可能都超过了（On），最后可以把list优化一下，把从 2^n个数据放进list里面变成一个方法，需要哪个求哪个。

        最后，虽然是写出了二分法，但是可能是示例太少，或者说考虑不周到，(log2n)的程序跑的巨慢