DEV Community

Abhishek Chaudhary
Abhishek Chaudhary

Posted on

N-ary Tree Level Order Traversal

Given an n-ary tree, return the level order traversal of its nodes' values.

Nary-Tree input serialization is represented in their level order traversal, each group of children is separated by the null value (See examples).

Example 1:

Input: root = [1,null,3,2,4,null,5,6]
Output: [[1],[3,2,4],[5,6]]

Example 2:

Input: root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
Output: [[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]

Constraints:

  • The height of the n-ary tree is less than or equal to 1000
  • The total number of nodes is between [0, 104]

SOLUTION:

"""
# Definition for a Node.
class Node:
    def __init__(self, val=None, children=None):
        self.val = val
        self.children = children
"""

class Solution:
    def levelOrder(self, root: 'Node') -> List[List[int]]:
        nodes = [(root, 0)]
        i = 0
        op = {}
        while i < len(nodes):
            curr, level = nodes[i]
            if curr:
                op[level] = op.get(level, []) + [curr.val]
                for node in curr.children:
                    nodes.append((node, level + 1))
            i += 1
        return list(op.values())
Enter fullscreen mode Exit fullscreen mode

Top comments (0)