1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
| package com.demo.s103;
import java.util.*;
class TreeNode { int val; TreeNode left; TreeNode right; TreeNode() {} TreeNode(int val) { this.val = val; } TreeNode(int val, TreeNode left, TreeNode right) { this.val = val; this.left = left; this.right = right; } } public class Solution { public List<List<Integer>> zigzagLevelOrder(TreeNode root) { List<List<Integer>> ret = new ArrayList<>(); if(root == null) { return ret; } Queue<TreeNode> queue = new LinkedList(); queue.offer(root); boolean zig = true; while(queue.size() != 0) { int size = queue.size(); Deque<Integer> level = new LinkedList(); for(int i = 0 ;i < size; i ++) { TreeNode cur = queue.poll(); if(zig) { level.offerLast(cur.val); } else { level.offerFirst(cur.val); } if(cur.left != null) { queue.offer(cur.left); } if(cur.right != null) { queue.offer(cur.right); }
} zig = !zig; ret.add(new LinkedList(level));
} return ret; } }
|