题目1
给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
解题思路:使用广度优先遍历,子循环使用层次节点数递减
代码如下:
public List<List<Integer>> levelOrder(TreeNode root) {
//广度优先遍历,加入所属层级
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(root == null) {
return res;
}
LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
while(!queue.isEmpty()) {
//每次遍历一层节点
List<Integer> list = new ArrayList<Integer>();
int count = queue.size();
while(count-- > 0) {
TreeNode cNode = queue.poll();
list.add(cNode.val);
if(cNode.left != null) {
queue.offer(cNode.left);
}
if(cNode.right != null) {
queue.offer(cNode.right);
}
}
res.add(list);
}
return res;
}
题目2:
给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层次遍历为:
[
[15,7],
[9,20],
[3]
]
代码实现
public List<List<Integer>> levelOrderBottom(TreeNode root) {
List<List<Integer>> list = new ArrayList<List<Integer>>();
if (root==null) {
return list;
}
TreeNode node = root;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(node);
int levelCount=1;
while(!queue.isEmpty()) {
List<Integer> level = new LinkedList<Integer>();
levelCount = queue.size();
while(levelCount -->0) {
TreeNode tmp = queue.poll();
level.add(tmp.val);
if (tmp.left!= null) {
queue.offer(tmp.left);
}
if(tmp.right!=null) {
queue.offer(tmp.right);
}
}
list.add(level);
}
Collections.reverse(list);
return list;
}