LeetCode429.N叉树的层序遍历

题目来源:

https://leetcode-cn.com/problems/n-ary-tree-level-order-traversal/

题目描述:

LeetCode429.N叉树的层序遍历

代码如下:

/*
// 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 {
	List<List<Integer>> list = new ArrayList<List<Integer>>();

	public List<List<Integer>> levelOrder(Node root) {
		levelOrder(0, root);
		return list;
	}

	public void levelOrder(int level, Node root) {
		if (root == null)
			return;

		if (list.size() == level) {
			list.add(new ArrayList<>());
		}

		list.get(level).add(root.val);
		for (Node node : root.children) {
			levelOrder(level + 1, node);
		}
	}
}