LeetCode429. N叉树的层序遍历

版权声明: https://blog.csdn.net/weixin_40550726/article/details/82947672

给定一个N叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。

例如,给定一个 3叉树 :

返回其层序遍历:

[
     [1],
     [3,2,4],
     [5,6]
]

说明:

  1. 树的深度不会超过 1000
  2. 树的节点总数不会超过 5000

思路:使用辅助数据结构queue。

/*
// Definition for a Node.
class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    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 LinkedList<List<Integer>>();
        if(null==root){
            return res;
        }
        ArrayDeque<Node> queue=new ArrayDeque<Node>();
        queue.add(root);
        while(!queue.isEmpty()){
            int size=queue.size();
            List<Integer> tmp=new LinkedList<Integer>();
            for(int i=0;i<size;i++){
                Node node=queue.removeFirst();
                tmp.add(node.val);
                for(Node n:node.children){
                    queue.addLast(n);
                }
            }
            res.add(tmp);
        }
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_40550726/article/details/82947672