题目描述:给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal
思路:
创建一个静态变量来存储每一层的元素。遍历过程中,若根节点为空,则直接返回result;若根节点不为空,则进入辅助方法。辅助方法help()本质是对这个树进行前序遍历,在递归遍历过程中,将每一层的元素存入每层对应的ArrayList中去,最后存入一个完整的List中。
注意:在线OJ 有多组例子,所以每次调用static都需要清空一次。
public List<List<Integer>> levelOrder(TreeNode root) {
//要把每一层的节点放到一个单独的list中,所以不能使用之前的层序遍历方法
// 还是通过递归来实现
//清空result,让它保持为空List
result.clear();
if (root==null){
return result;
}
help(root,0);
return result;
}
public static void help(TreeNode root,int level){
//给level对应的层里添加ArrayList
if (level==result.size()){
result.add(new ArrayList<>());//防止下标越界情况
}
//把当前节点添加到arrayList的合适位置
result.get(level).add(root.val);
if (root.left!=null){
//继续向下添加,更新level
help(root.left,level+1);
}
if (root.right!=null){
help(root.right,level+1);
}
}