【LeetCode】N叉树遍历

题目描述

给定一个 n 叉树的根节点 root ，返回 其节点值的 前序遍历 。

n 叉树 在输入中按层序遍历进行序列化表示，每组子节点由空值 null 分隔（请参见示例）。

示例

示例 1：

输入：root = [1,null,3,2,4,null,5,6]
输出：[1,3,5,6,2,4]

方法

N叉树遍历相对于二叉树遍历，我们要考虑他的N个孩子，因此引入for循环遍历访问N个孩子即可。

总代码

• 二叉树遍历

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def preorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        ans = []
        def mypreorder(node: Optional[TreeNode]):
            if node != None:
                ans.append(node.val)
                mypreorder(node.left)
                mypreorder(node.right)
        mypreorder(root)
        return ans

• N叉树遍历

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

class Solution:
    def preorder(self, root: 'Node') -> List[int]:
        ans = []
        def dfs(node: 'Node'):
            if node is None:
                return
            ans.append(node.val)
            for ch in node.children:
                dfs(ch)
        dfs(root)
        return ans