题目描述
Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example:
Given binary tree {3,9,20,#,#,15,7}
,
3 / \ 9 20 / \ 15 7
return its level order traversal as:
[ [3], [9,20], [15,7] ]按层输出二叉树。
解题思路
采用队列,逐层添加到队列中,每次将队首的元素取出并将其孩子节点添加到队尾,直到队列为空为止。
代码
/**
* 按层输出二叉树
* @param root
* @return
*/
public static List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> levelList = new ArrayList<List<Integer>>();
if (root == null)
return levelList;
Queue<TreeNode> queue = new LinkedList<TreeNode>();//队列
TreeNode temp;
int lastLevelLen = 1;//上一层元素个数
int currentLevelLen = 0;//当前层元素个数
List<Integer> list = new ArrayList<Integer>();
queue.add(root);//添加根节点到队列
while (!queue.isEmpty()) {
temp = queue.poll();
list.add(temp.val);
lastLevelLen--;
if (temp.left != null) {
//如果左孩子不为空,将其添加到队尾,当前层节点个数加一
queue.offer(temp.left);
currentLevelLen++;
}
if (temp.right != null) {
//如果右孩子不为空,将其添加到队尾,当前层节点个数加一
queue.offer(temp.right);
currentLevelLen++;
}
if (lastLevelLen == 0) {
//如果上一层已经遍历完毕,且已将上一层的孩子节点添加到队列里
lastLevelLen = currentLevelLen;
currentLevelLen = 0;
levelList.add(list);
list = new ArrayList<Integer>();
}
}
return levelList;
}