1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
| class Solution { public int countNodes(TreeNode root) { int count = 0; if(root == null){ return count; } LinkedList<TreeNode> queue = new LinkedList<>(); queue.offer(root); while(!queue.isEmpty()){ int n = queue.size(); for(int i = 0; i < n; i++){ TreeNode node = queue.poll(); count++; if(node.left != null){ queue.offer(node.left); } if(node.right != null){ queue.offer(node.right); } } } return count; } }
|