102. 二叉樹的層序遍歷
很巧妙的,又學習了一種層次遍歷的方法,就是說根據當前的隊列的長度去遍歷,遍歷的當前隊列的長度就是該層次的節點個數。
/*** 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;* }* }*/
class Solution {public List<List<Integer>> levelOrder(TreeNode root) {List<List<Integer>> levelOrderList = new ArrayList<>();if(root == null) return levelOrderList;Queue<TreeNode> queue = new LinkedList<>(); //用于做隊列,遍歷整棵樹List<Integer> list = new ArrayList<>(); // 用于存儲用隊列遍歷樹的valqueue.add(root);while(queue.size()!=0) {// System.out.println("當前隊列的長度" + queue.size());int l = queue.size();for(int i=0;i<l; i++) {TreeNode front = queue.poll(); //LinkedList的poll方法可以取出第一個元素的值。list.add(front.val);if(front.left!=null) queue.add(front.left);if(front.right!=null) queue.add(front.right);}
/* System.out.print("當前放入的元素為");Iterator iterator = list.iterator();while(iterator.hasNext()){System.out.print(iterator.next() + " ");}System.out.println(); */levelOrderList.add(list);list = new ArrayList<Integer>();}return levelOrderList;}
}