102. 二叉树的层次遍历

Posted 2020-11-24 zccfrancis

给定一个二叉树，返回其按层次遍历的节点值。 （即逐层地，从左到右访问所有节点）。

例如:
给定二叉树: [3,9,20,null,null,15,7],

3
/
9 20
/
15 7
返回其层次遍历结果：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

DFS通过创建levels数组，记录level来实现层次遍历。

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode(int x) { val = x; }
 8  * }
 9  */
10 class Solution {
11     List<List<Integer>> levels = new ArrayList<List<Integer>>();
12 
13     public void DFS(TreeNode node, int level) {
14         if (levels.size() <= level) {
15             levels.add(new ArrayList<>());
16         }
17 
18         levels.get(level).add(node.val);
19         if (node.left != null)
20             DFS(node.left, level + 1);
21         if (node.right != null)
22             DFS(node.right, level + 1);
23     }
24     
25     public List<List<Integer>> levelOrder(TreeNode root) {
26         if (root == null) return levels;
27         DFS(root, 0);
28         return levels;
29     }
30 }