hot100-二叉树的层序遍历

二叉树的层序遍历

题目描述：

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

思路：

我们的目的是将同一层的节点值加到各自的list中，但是并不一定要遍历时也按照这个顺序去一个个的遍历节点。

那么给每层的节点一个level标记即可，比如root的level为0，我们的目的是节点的level为几，就将其加入res中第几个list。

只需要在递归时，每次处理子节点，递归函数中level就加一即可，这样就可以保证处理某个节点时，传下来的参数level就是深度。

需要注意，res最开始是为空的，当我们发现下标为level的list还不存在于res中时，我们就在res中新增一个list，这样可以防止下标溢出。关键代码如下：

if(res.size()<level+1){
    res.add(new ArrayList<>());
}
res.get(level).add(node.val);

代码：

public static List<List<Integer>> res = new ArrayList<>();
public static List<List<Integer>> levelOrder(TreeNode2 root) {
    res.clear();
    fun(root,0);
    return res;
}
public static void fun(TreeNode2 node,int level){
    if(node==null)return;
    if(res.size()<level+1){
        res.add(new ArrayList<>());
    }
    res.get(level).add(node.val);
    fun(node.left,level+1);
    fun(node.right,level+1);
}