【leetcode】429. N 叉树的层序遍历

Fancier__

已于 2024-03-09 22:24:03 修改

阅读量431

点赞数 6

文章标签： leetcode 深度优先算法

于 2024-03-09 22:18:46 首次发布

本文链接：https://blog.csdn.net/hhh___fff113/article/details/136592191

版权

题目描述

给定一个 N 叉树，返回其节点值的_层序遍历_。（即从左到右，逐层遍历）。

树的序列化输入是用层序遍历，每组子节点都由 null 值分隔（参见示例）。

示例 1：

|left|200

输入： root = [1,null,3,2,4,null,5,6]
输出： [[1],[3,2,4],[5,6]]

示例 2：

输入： root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出： [[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]

提示：

树的高度不会超过 1000
树的节点总数在 [0, 10^4] 之间

代码

bfs

class Solution {  
    public List<List<Integer>> levelOrder(Node root) {  
        Queue<Node> queue = new LinkedList<>();  
        ArrayList<List<Integer>> lists = new ArrayList<>();  
        if (root == null) return lists;  
        lists.add(new ArrayList<Integer>(){{add(root.val);}});  
        for (Node child : root.children) {  
            queue.add(child);  
        }  
        while (!queue.isEmpty()) {  
            int len = queue.size();  
            ArrayList<Integer> temp = new ArrayList<>();  
            for (int i = 0; i < len; i++) {  
                Node poll = queue.poll();  
                temp.add(poll.val);  
                for (Node child : poll.children) {  
                    queue.add(child);  
                }  
            }  
            lists.add(temp);  
        }  
        return lists;  
    }  
}

dfs

class Solution {  
    List<List<Integer>> ans = new ArrayList<>();  
    public List<List<Integer>> levelOrder(Node root) {  
        dfs(root, 1);  
        return ans;  
    }  
    private void dfs(Node node, int deep) {  
        if (node == null) return;  
        if (ans.size() < deep) {  
            ans.add(new ArrayList<>());  
        }  
        ans.get(deep - 1).add(node.val);  
        for (Node child : node.children) {  
            dfs(child, deep + 1);  
        }  
    }  
}