day-23 N 叉树的层序遍历

思路:利用bfs,利用队列将当前层的孩子元素加入队列,再将本层元素出队即可

注意点:res.remove()执行后,ArrayList的元素索引会改变,可以先添加下一层所有元素后,再将本层元素一起出队

code:

java 复制代码
/*
// Definition for a Node.
class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> res=new ArrayList<List<Integer>>();
        if(root==null)return res;
        List<Node> now=new ArrayList<Node>();
        now.add(root);
        while(now.size()>0){
            int num=now.size();
            List<Integer> t=new ArrayList<Integer>();
            for(int i=0;i<num;i++){//遍历同一层元素
                t.add(now.get(i).val);
                int cur=now.get(i).children.size();
                for(int j=0;j<cur;j++){//遍历每个元素的孩子
                    if(now.get(i).children.get(j)!=null){
                        now.add(now.get(i).children.get(j));
                    }
                }
            }
            for(int i=0;i<num;i++){
                now.remove(0);
            }
            res.add(t);
        }
        return res;
    }
}
相关推荐
梓䈑5 小时前
【算法题攻略】BFS 解决FloodFill 算法、最短路问题、多源BFS 和 拓扑排序
c++·算法·宽度优先
Adios7948 小时前
设置交集大小至少为2
数据结构·算法·leetcode
来一碗刘肉面14 小时前
栈的应用(表达式求值)
数据结构·算法
xiaowang1234shs15 小时前
怪兽轻断食技术深度测评:从断食计时引擎到AI识别算法的工程实践解析
数据库·人工智能·算法·macos·机器学习·p2p·visual studio
小果因子实验室15 小时前
量化研究--策略迁移算法1研究
算法
Web极客码16 小时前
如何用三段式确定性剪枝,为 LLM Agent 砍掉 35% 的 Token 成本?
服务器·人工智能·算法·机器学习
程序猿乐锅16 小时前
【数据结构与算法 | 第六篇】力扣1109,1094差分数组
java·算法·leetcode
wabs66616 小时前
关于图论【广度优先搜索的理论基础】
算法·图论·宽度优先
孤魂23316 小时前
逻辑回归算法
算法·机器学习·逻辑回归