力扣练习日记——二叉树的层序遍历

二叉树的层序遍历

先放题目
给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

示例：
二叉树：[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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

深度遍历

这里先放代码

public class LevelOrder {
    private static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) { val = x; }
    }
    
    private static List<List<Integer>> levelOrder(TreeNode root){
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) return res;
        dfs(1, root, res);
        return res;
    }

    private static void dfs(int index, TreeNode root, List<List<Integer>> res){
        if (res.size() < index) res.add(new ArrayList<>());
        res.get(index-1).add(root.val);
        if (root.left != null){
            dfs(index + 1, root.left, res);
        }
        if (root.right != null){
            dfs(index + 1, root.right, res);
        }
    }

    public static void main(String[] args) {
        TreeNode root = new TreeNode(3);
        root.left = new TreeNode(9);
        root.right = new TreeNode(20);
        root.right.left = new TreeNode(15);
        root.right.right = new TreeNode(7);
        for (List<Integer> integers : levelOrder(root)) {
            for (Integer integer : integers) {
                System.out.println(integer);
            }
            System.out.println("\n");;
        }
    }
}

这里采用深度遍历解决问题，先深度遍历，顺序为3->9->20->15->7，index代表层数，每深入新的一层，index加一，用index对应到每层的子节点，遍历，完事儿。