[leetcode] Binary Tree Level Order Traversal

Posted 2021-01-12 Lin.B

Binary Tree Level Order Traversal Ⅰ

Given a binary tree, return the level order traversal of its nodes‘ values. (ie, from left to right, level by level).

For example:
Given binary tree [3,9,20,null,null,15,7],

    3
   /   9  20
    /     15   7

 

return its level order traversal as:

[
  [3],
  [9,20],
  [15,7]
]

---

分析：题目翻译一下：对一个二叉树进行层次遍历，BFS。我比较喜欢用队列来做，BFS比较基础了，也很简单。

代码如下：

 1 class Solution {
 2     public List<List<Integer>> levelOrder(TreeNode root) {
 3         List<List<Integer>> res = new ArrayList<>();
 4         if ( root == null ) return res;
 5         Queue<TreeNode> queue = new LinkedList<>();
 6         queue.offer(root);
 7         while ( !queue.isEmpty() ){
 8             int size = queue.size();
 9             List<Integer> list = new ArrayList<>();
10             for ( int i = 0 ; i < size ; i ++ ){
11                 TreeNode t = queue.poll();
12                 list.add(t.val);
13                 if ( t.left != null )  queue.offer(t.left);
14                 if ( t.right != null ) queue.offer(t.right);
15             }
16             res.add(list);
17         }
18         return res;
19     }
20 }

    运行时间3ms。

 

Binary Tree Level Order Traversal II

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 [3,9,20,null,null,15,7],

    3
   /   9  20
    /     15   7

 

return its bottom-up level order traversal as:

[
  [15,7],
  [9,20],
  [3]
]

---

分析：题目翻译一下：倒着层次遍历一颗二叉树

思路1：根据上面的案例，也就是将BFS结果倒着输出就可以了。

这里考察一个只是点就是如何将list倒序，使用Collections.reverse(list)这个方法。注：Collections这个类里面有很多对于list、map这些集合对象的方法。有空可以查一下。

 1 public List<List<Integer>> levelOrder(TreeNode root) {
 2         List<List<Integer>> res = new ArrayList<>();
 3         Queue<TreeNode> queue = new LinkedList<>();
 4         queue.offer(root);
 5         while ( !queue.isEmpty() ){
 6             int size = queue.size();
 7             List<Integer> list = new ArrayList<>();
 8             for ( int i = 0 ; i < size ; i ++ ){
 9                 TreeNode t = queue.poll();
10                 list.add(t.val);
11                 if ( t.left != null )  queue.offer(t.left);
12                 if ( t.right != null ) queue.offer(t.right);
13             }
14             res.add(list);
15         }
16         Collections.reverse(res);
17         return res;
18     }

思路二：将层的结果放入到一个stack中，最后读取stack中的数据也可以实现倒序。就不写代码了。