算法学习：222.完全二叉树的节点个数

完全二叉树的节点个数

题目链接：力扣题目链接
难度：中等
给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。
完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2^h 个节点。
示例
!转载力扣官网](https://img-blog.csdnimg.cn/img_convert/509c4145eead75a3dac44e3408cae381.png#pic_center)
输入：root = [1,2,3,4,5,6]
输出：6

思路

按照普通二叉树的逻辑来求

二叉树节点的定义

 //二叉树节点的定义。
 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;
     }
}

递归代码

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

迭代代码

class Solution{
	public int countNodes(TreeNode root){
		if(root == null){
			return 0;
		}
		Queue<TreeNode> queue = new LinkedList<>();
		queue.offer(root);
		int result = 0;
		while(!queue.isEmpty()){
			int len = queue.size();
			for(int i=0;i<len;i++){
				TreeNode node = queue.poll();
				result++;
				if(node.left != null)	queue.offer(node.left);
				if(node.right != null)	queue.offer(node.right);
			}
		}
		return result;
	}
}