問題描述

給你一個二叉樹，請你返回其按 層序遍歷 得到的節點值。 （即逐層地，從左到右訪問所有節點）。

示例：
二叉樹：[3,9,20,null,null,15,7],

返回其層次遍歷結果：

[
[3],
[9,20],
[15,7]
]

問題解法

先構建廣度優先搜尋，再設定一個指標標誌這一層有多少個節點。

/**
 * 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) {
		 if(root==null)
			 return new ArrayList<List<Integer>>();
	        Queue<TreeNode> q=new LinkedList<TreeNode>();
	        int end=1;
	        TreeNode p=root;
	        q.add(p);
	        ArrayList<List<Integer>> list1=new ArrayList<List<Integer>>();
	        ArrayList<Integer> list=new ArrayList<Integer>();
	        while(!q.isEmpty()) {
	        	p=q.poll();
	        	if(p.left!=null)
	        		q.add(p.left);
	        	if(p.right!=null)
	        		q.add(p.right);
	        	list.add(p.val);
	        	end--;
	        	if(end==0) {
	        		end=q.size();
	        		list1.add(list);
	        		list=new ArrayList<Integer>() ;
	        	}
	        }
	        return list1;
	    }
}
 

執行結果