一般二叉树的遍历都是用递归的方式
一开始没用递归,使用双栈进行遍历,同时为了栈读取效率更高,自己写了个数组和下标指针来模拟栈。
通过后,看题解,发现不用一层层去遍历,因为每一层的List都是独立的(List<List< Integer>>),只要正常dfs的同时,判别所处深度,修改对应深度的 List 即可。
自己的(1ms)
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> ans = new ArrayList<>();
if(root == null)
return ans;
TreeNode[][] t = new TreeNode[2][1000];
t[0][0] = root;
int[] index = new int[2];
index[0] = 1;
for(int i = 0;;i++){
List<Integer> l = new ArrayList<>();
for(;index[i % 2] > 0; index[i % 2]--){
if(t[i % 2][index[i % 2] - 1] == null)
continue;
l.add(t[i % 2][index[i % 2] - 1].val);
if(i % 2 == 0){
t[1][index[1]++] = t[0][index[0] - 1].left;
t[1][index[1]++] = t[0][index[0] - 1].right;
}
else{
t[0][index[0]++] = t[1][index[1] - 1].right;
t[0][index[0]++] = t[1][index[1] - 1].left;
}
}
if(index[0] == 0 && index[1] == 0)
break;
ans.add(l);
}
return ans;
}
}
题解(0ms)
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> records = new ArrayList<>(0);
fn(root, records, 0);
return records;
}
public void fn(TreeNode tn, List<List<Integer>> records, int level) {
if(tn == null) {
return;
}
if(records.size() <= level){
records.add(new ArrayList<Integer>());
}
List<Integer> current = records.get(level);
if((level & 1) == 0){
current.add(tn.val);
} else {
current.add(0, tn.val);
}
fn(tn.left, records, level + 1);
fn(tn.right, records, level + 1);
}
}