题目链接
算法思路
BFS算法框架
这题主要考查的是树的层次遍历,只要按层次遍历思路来遍历就行了!唯一不同是在添加到结果列表的时候,先判断一下是否要反转列表。通过奇偶数来判断是否要反转列表。
代码实现
import java.util.*;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
//结果集列表
ArrayList<ArrayList<Integer>> res = new ArrayList<>();
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
if (pRoot == null) {
return res;
}
levelTravers(pRoot);
return res;
}
//层次遍历
private void levelTravers(TreeNode pRoot) {
Queue<TreeNode> q = new LinkedList<>();
q.offer(pRoot);
while (!q.isEmpty()) {
int size = q.size();
ArrayList<Integer> list = new ArrayList<>();
for (int i = 0; i < size; i++) {
TreeNode node = q.poll();
list.add(node.val);
if(node.left!=null){
q.offer(node.left);
}
if (node.right != null) {
q.offer(node.right);
}
}
//如果是奇数,则反转list
if((res.size() & 1) == 1 ){
Collections.reverse(list);
}
res.add(list);
}
}
}