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
| package com.demo.s102;
import java.util.ArrayList; import java.util.LinkedList; import java.util.List; import java.util.Queue;
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>> levelOrder(TreeNode root) { List<List<Integer>> ret = new ArrayList(); if(root == null) { return ret; }
Queue<TreeNode> queue = new LinkedList<TreeNode>(); queue.offer(root); while(queue.size() != 0) { List<Integer> level = new ArrayList<Integer>(); int size = queue.size(); for(int i = 0; i< size; i++) { TreeNode cur = queue.poll(); level.add(cur.val); if(cur.left != null) { queue.offer(cur.left); } if(cur.right != null) { queue.offer(cur.right); } } ret.add(level);
} return ret;
} }
|