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;
    }
}
相关推荐
じ☆冷颜〃3 小时前
黎曼几何驱动的算法与系统设计:理论、实践与跨领域应用
笔记·python·深度学习·网络协议·算法·机器学习
数据大魔方3 小时前
【期货量化实战】日内动量策略:顺势而为的短线交易法(Python源码)
开发语言·数据库·python·mysql·算法·github·程序员创富
POLITE33 小时前
Leetcode 23. 合并 K 个升序链表 (Day 12)
算法·leetcode·链表
楚来客4 小时前
AI基础概念之八:Transformer算法通俗解析
人工智能·算法·transformer
Echo_NGC22374 小时前
【神经视频编解码NVC】传统神经视频编解码完全指南:从零读懂 AI 视频压缩的基石
人工智能·深度学习·算法·机器学习·视频编解码
会员果汁4 小时前
leetcode-动态规划-买卖股票
算法·leetcode·动态规划
橘颂TA5 小时前
【剑斩OFFER】算法的暴力美学——二进制求和
算法·leetcode·哈希算法·散列表·结构与算法
地平线开发者6 小时前
征程 6 | cgroup sample
算法·自动驾驶
姓蔡小朋友7 小时前
算法-滑动窗口
算法