【Leetcode】429. N-ary Tree Level Order Traversal

最新推荐文章于 2023-11-08 11:57:21 发布

原创最新推荐文章于 2023-11-08 11:57:21 发布 · 86 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种解决N叉树层次遍历问题的方法，通过使用递归算法，实现了从左到右，按层级返回树节点值的功能。文章详细解释了如何通过检查当前层级与结果集大小的关系来确定是否需要新增一层ArrayList，以及如何遍历N叉树的子节点。

题目：

Given an n-ary tree, return the level order traversal of its nodes’ values. (ie, from left to right, level by level).

For example, given a 3-ary tree:
在这里插入图片描述注：好像如果是正常的树，就是root.left 和 root.righ。如果是这种就是写个for循环用root.children遍历。
这里想了半天：如果需要用到新加一层arraylist()
if(level==res.size()){
res.add(new ArrayList());
}

Code：（个人比较喜欢递归

class Solution {

List<List<Integer>> res = new ArrayList<List<Integer>>();
public List<List<Integer>> levelOrder(Node root) {    
    bfs(root,0);
    return res;
}
public void bfs(Node root, int level){
    if(root==null)
        return ;
    if(level==res.size()){
        res.add(new ArrayList());
    } 
    res.get(level).add(root.val);
    for(Node node : root.children)
        bfs(node, level+1);
}

}