給定一個二叉樹,返回其按層次遍歷的節點值。 (即逐層地,從左到右訪問所有節點)。
思路:可以利用隊列先進先出的特性,將將二叉樹中的節點按層次放進隊列中,然後再依次出隊列。
具體代碼如下:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ret = new ArrayList<>();
if(root==null) {
return ret;
}
Queue<TreeNode> q = new LinkedList<>();
q.offer(root); //先將根節點放入隊列中
while(!q.isEmpty()) { //然後開始循環遍歷
int size = q.size(); //size爲當前層的元素個數
List<Integer> level = new ArrayList<>(size);//用level來保存一層元素
//一次性遍歷一層的元素
for(int i=0;i<size;i++) {
TreeNode cur = q.poll(); //將當前層節點依次出隊列
level.add(cur.val); //將當前層節點值放進level中
//如果節點有左孩子,就將左孩子入隊列
if(cur.left != null) {
q.offer(cur.left);
}
//如果節點有右孩子,就將右孩子入隊列
if(cur.right != null) {
q.offer(cur.right);
}
}
ret.add(level); //每遍歷完一層就將當前層結果放入ret中
}
return ret;
}
}