LeetCode 429. N 叉树的层序遍历

429. N 叉树的层序遍历

思路： 图的层次遍历时，一般用队列进行逐层遍历。当需要记录当前层有哪些元素时，可以使用一层循环访问当前队列的元素。
看代码应该一下就懂了

/*
// 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>> ans = new ArrayList<>();
        Deque<Node> q = new LinkedList<>();
        if(root == null) return ans ;
        q.add(root);
        while(!q.isEmpty()){
            List<Integer> temp = new ArrayList<>(); //  存储当前层的元素
            int size = q.size();  //  *  记录当前层有几个元素
            for(int i = 0  ; i < size ; i ++){   //  访问当前层的元素
                Node t = q.poll();   //  元素出队
                for(Node node : t.children){
                    q.add(node);  //  添加子节点
                }
                temp.add(t.val); //  记录当前层元素
            }
            ans.add(temp);  
        }
        return ans ;
    }
}