求二叉树结点个数

给定一棵非空二叉树,求该二叉树的节点个数
输入、输出描述
输入:
给定一个二叉树的根节点
输出:
该二叉树的节点个数
Example
输入:
二叉树如下:
    1
   / \
   2  3
  /  / \
  4  5  6
输出:
6
代码:
import java.util.*;

public class Main {

    /**
    //该段代码仅用于调试,提交时请注释该段代码
    class TreeNode<T> {

        public T data;

        public TreeNode<T> left;

        public TreeNode<T> right;
    }
 */ 

 public int solution(TreeNode<Integer> root) {
	if(root ==null){
		
		return 0;
	}
	 List<Integer> list = new ArrayList<>();
	 
	Queue<TreeNode<Integer>> q = new LinkedList<>();
	TreeNode<Integer>p=root;
	q.add(p);
	while (q.isEmpty() == false)
	{	
		p=q.poll();
		list.add(p.data);
		if (p.left != null)
		{
			q.add(p.left);
		}
		if (p.right != null)
		{
			q.add(p.right);
		}
	}
    return list.size();
}
}
一个创业中的苦逼程序员
评论专区

隐藏