Given a binary tree, return the bottom-up level order traversal of its nodes' values. (ie, from left to right, level by level from leaf to root).
For example:
Given binary tree
Given binary tree
{3,9,20,#,#,15,7}
,3 / \ 9 20 / \ 15 7
return its bottom-up level order traversal as:
[ [15,7], [9,20], [3] ]Solution: BFS
和Binary Tree Level Order Traversal一样,最后reverse一下list得到bottom-up结果.
时间上和空间上仍是O(n)。
/** * Definition for binary tree * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ public class Solution { public ArrayList> levelOrderBottom(TreeNode root) { ArrayList > result = new ArrayList >(); if(root == null) return result; LinkedList queue = new LinkedList (); queue.add(root); int curLevCount = 1; int nextLevCount = 0; ArrayList levelRes = new ArrayList (); ArrayList > list = new ArrayList >(); while(!queue.isEmpty()) { TreeNode cur = queue.poll(); curLevCount--; levelRes.add(cur.val); if(cur.left != null) { queue.add(cur.left); nextLevCount++; } if(cur.right != null) { queue.add(cur.right); nextLevCount++; } if(curLevCount == 0) { curLevCount = nextLevCount; nextLevCount = 0; list.add(levelRes); levelRes = new ArrayList (); } } //reverse the list for(int i = list.size() - 1; i >= 0; i--) { result.add(list.get(i)); } return result; } }
No comments:
Post a Comment