Java 完全二叉树的节点个数

完全二叉树的节点个数

中等

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

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

示例 1：

输入：root = [1,2,3,4,5,6]

输出：6

示例 2：

输入：root = []

输出：0

示例 3：

输入：root = [1]

输出：1

题解

1. 构建入例一所示完全二叉树，num 记录树节点个数

2. 创建一个countNodes()方法，传入根节点参数，当树为空时返回0，树只有一个根节点时返回1。否则调用infixOrder中序遍历方法传入root根节点。

3. 创建中序遍历方法infixOrder(),遍历顺序（左中右），如果传入节点左子树不为空，递归左子树，num++,在判断右子树不为空，递归右子树。

4. 接收countNodes方法返回值输出，结果就是树节点个数。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
package lanqiaobei.com.yjm.Medium;
// 完全二叉树结点个数
public class WanQuanErChaShuJieDianGeShu {
    private static int num;
    public static void main(String[] args) {
        TreeNode  root = new TreeNode(1);
        TreeNode  node1 = new TreeNode(2);
        TreeNode  node2 = new TreeNode(3);
        TreeNode  node3 = new TreeNode(4);
        TreeNode  node4 = new TreeNode(5);
        TreeNode  node5 = new TreeNode(6);
        root.left = node1;
        root.right = node2;
        node2.left = node3;
        node2.right = node4;
        node3.left = node5;
        System.out.println(countNodes(root));
    }
    public static int countNodes(TreeNode root) {
        if(root == null){
            return 0;
        }
        if(root.left == null && root.right == null){
            return 1;
        }
        infixOrder(root);
        return num;
    }
    public static void infixOrder(TreeNode root){
        if(root.left != null){
            infixOrder(root.left);
        }
        num++;
        if(root.right != null){
            infixOrder(root.right);
        }
    }
}