update 0222.完全二叉树的节点个数: 替换 java 代码，调整 js 格式

problems/0222.完全二叉树的节点个数.md

@@ -63,7 +63,7 @@ int getNodesNum(TreeNode* cur) {
 if (cur == NULL) return 0;
 ```
 
-3. 确定单层递归的逻辑：先求它的左子树的节点数量，再求的右子树的节点数量，最后取总和再加一 （加1是因为算上当前中间节点）就是目前节点为根节点的节点数量。
+3. 确定单层递归的逻辑：先求它的左子树的节点数量，再求右子树的节点数量，最后取总和再加一 （加1是因为算上当前中间节点）就是目前节点为根节点的节点数量。
 
 代码如下：
 
@@ -168,7 +168,7 @@ public:
 
 可以看出如果整个树不是满二叉树，就递归其左右孩子，直到遇到满二叉树为止，用公式计算这个子树（满二叉树）的节点数量。
 
-这里关键在于如果去判断一个左子树或者右子树是不是满二叉树呢？ 
+这里关键在于如何去判断一个左子树或者右子树是不是满二叉树呢？ 
 
 在完全二叉树中，如果递归向左遍历的深度等于递归向右遍历的深度，那说明就是满二叉树。如图： 
 
@@ -178,13 +178,13 @@ public:
 
 ![](https://code-thinking-1253855093.file.myqcloud.com/pics/20220829163709.png) 
 
-哪有录友说了，这种情况，递归向左遍历的深度等于递归向右遍历的深度，但也不是满二叉树，如题： 
+那有录友说了，这种情况，递归向左遍历的深度等于递归向右遍历的深度，但也不是满二叉树，如题： 
 
 ![](https://code-thinking-1253855093.file.myqcloud.com/pics/20220829163811.png) 
 
 如果这么想，大家就是对 完全二叉树理解有误区了，**以上这棵二叉树，它根本就不是一个完全二叉树**！ 
 
-判断其子树岂不是满二叉树，如果是则利用用公式计算这个子树（满二叉树）的节点数量，如果不是则继续递归，那么 在递归三部曲中，第二部：终止条件的写法应该是这样的： 
+判断其子树是不是满二叉树，如果是则利用公式计算这个子树（满二叉树）的节点数量，如果不是则继续递归，那么 在递归三部曲中，第二部：终止条件的写法应该是这样的： 
 
 ```CPP
 if (root == nullptr) return 0; 
@@ -292,26 +292,22 @@ class Solution {
      * 满二叉树的结点数为：2^depth - 1
      */
     public int countNodes(TreeNode root) {
-        if(root == null) {
-            return 0;
+        if (root == null) return 0;
+        TreeNode left = root.left;
+        TreeNode right = root.right;
+        int leftDepth = 0, rightDepth = 0; // 这里初始为0是有目的的，为了下面求指数方便
+        while (left != null) {  // 求左子树深度
+            left = left.left;
+            leftDepth++;
         }
-        int leftDepth = getDepth(root.left);
-        int rightDepth = getDepth(root.right);
-        if (leftDepth == rightDepth) {// 左子树是满二叉树
-            // 2^leftDepth其实是 （2^leftDepth - 1） + 1 ，左子树 + 根结点
-            return (1 << leftDepth) + countNodes(root.right);
-        } else {// 右子树是满二叉树
-            return (1 << rightDepth) + countNodes(root.left);
+        while (right != null) { // 求右子树深度
+            right = right.right;
+            rightDepth++;
         }
-    }
-
-    private int getDepth(TreeNode root) {
-        int depth = 0;
-        while (root != null) {
-            root = root.left;
-            depth++;
+        if (leftDepth == rightDepth) {
+            return (2 << leftDepth) - 1; // 注意(2<<1) 相当于2^2，所以leftDepth初始为0
         }
-        return depth;
+        return countNodes(root.left) + countNodes(root.right) + 1;
     }
 }
 ```
@@ -397,7 +393,7 @@ class Solution:
  *     Right *TreeNode
  * }
  */
-//本题直接就是求有多少个节点，无脑存进数组算长度就行了。
+//本题直接就是求有多少个节点，无脑存进结果变量就行了。
 func countNodes(root *TreeNode) int {
     if root == nil {
         return 0
@@ -473,15 +469,15 @@ func countNodes(root *TreeNode) int {
 var countNodes = function(root) {
     //递归法计算二叉树节点数
     // 1. 确定递归函数参数
-    const getNodeSum=function(node){
+    const getNodeSum = function(node) {
     //2. 确定终止条件
-        if(node===null){
+        if(node === null) {
             return 0;
         }
     //3. 确定单层递归逻辑
-        let leftNum=getNodeSum(node.left);
-        let rightNum=getNodeSum(node.right);
-        return leftNum+rightNum+1;
+        let leftNum = getNodeSum(node.left);
+        let rightNum = getNodeSum(node.right);
+        return leftNum + rightNum + 1;
     }
     return getNodeSum(root);
 };
@@ -491,19 +487,19 @@ var countNodes = function(root) {
 ```javascript
 var countNodes = function(root) {
     //层序遍历
-    let queue=[];
-    if(root===null){
+    let queue = [];
+    if(root === null) {
         return 0;
     }
     queue.push(root);
-    let nodeNums=0;
-    while(queue.length){
-        let length=queue.length;
-        while(length--){
-            let node=queue.shift();
+    let nodeNums = 0;
+    while(queue.length) {
+        let length = queue.length;
+        while(length--) {
+            let node = queue.shift();
             nodeNums++;
-            node.left&&queue.push(node.left);
-            node.right&&queue.push(node.right);
+            node.left && queue.push(node.left);
+            node.right && queue.push(node.right);
         }
     }
     return nodeNums;
@@ -514,24 +510,24 @@ var countNodes = function(root) {
 ```javascript
 var countNodes = function(root) {
     //利用完全二叉树的特点
-    if(root===null){
+    if(root === null) {
         return 0;
     }
-    let left=root.left;
-    let right=root.right;
-    let leftDepth=0,rightDepth=0;
-    while(left){
-        left=left.left;
+    let left = root.left;
+    let right = root.right;
+    let leftDepth = 0, rightDepth = 0;
+    while(left) {
+        left = left.left;
         leftDepth++;
     }
-    while(right){
-        right=right.right;
+    while(right) {
+        right = right.right;
         rightDepth++;
     }
-    if(leftDepth==rightDepth){
-        return Math.pow(2,leftDepth+1)-1;
+    if(leftDepth == rightDepth) {
+        return Math.pow(2, leftDepth+1) - 1;
     }
-    return countNodes(root.left)+countNodes(root.right)+1;
+    return countNodes(root.left) + countNodes(root.right) + 1;
 };
 ```