基本思路
难点在于不同层怎么按不同顺序打印节点值,不同层可以通过记录层数值判断是奇数层还是偶数层,打印节点值首先想的是如果偶数层就将队列中的下一层节点顺序反序,但是在for循环中加入下一层的节点一直是按左子树到右子树的顺序进行的,如果下一层节点的正确顺序为[5,7,9,11],反转前队列中节点为[9,11,5,7],反转后[7,5,11,9],还是不对。
正确做法是判断不同层后,将当前层数组保存的节点值反序就行了,因为数组中是从左到右的顺序存储节点,反转后是从右到左存储节点。
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return int整型ArrayList<ArrayList<>>
*/
public ArrayList<ArrayList<Integer>> Print (TreeNode pRoot) {
// write code here
ArrayList<ArrayList<Integer>> ret = new ArrayList<>();
if (pRoot == null) {
return ret;
}
int index = 1;
Queue<TreeNode> q = new ArrayDeque<>();
q.offer(pRoot);
while(!q.isEmpty()) {
int n = q.size();
ArrayList<Integer> row = new ArrayList<>();
index++;
for (int i = 0; i < n; ++i) {
TreeNode current = q.poll();
row.add(current.val);
if (current.left != null) {
q.offer(current.left);
}
if (current.right != null) {
q.offer(current.right);
}
}
if (index % 2 != 0) {
Collections.reverse(row);
}
ret.add(row);
}
return ret;
}
}



京公网安备 11010502036488号