leetcode 589/590. N叉树的前/后序遍历(python)

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

 

例如，给定一个 3叉树 :

 

 

返回其前序遍历: [1,3,5,6,2,4]。

 

说明: 递归法很简单，你可以使用迭代法完成此题吗?

"""
# Definition for a Node.
class Node(object):
    def __init__(self, val, children):
        self.val = val
        self.children = children
"""
class Solution(object):
    def preorder(self, root):
        """
        :type root: Node
        :rtype: List[int]
        """
        if not root:
            return [];
        if not root.children:
            return [root.val];
        result = [root.val];
        for child in root.children:
            result += self.preorder(child);
        return result;

 

给定一个N叉树，返回其节点值的后序遍历。

 

例如，给定一个 3叉树 :

 

 

返回其后序遍历: [5,6,3,2,4,1].

"""
# Definition for a Node.
class Node(object):
    def __init__(self, val, children):
        self.val = val
        self.children = children
"""
class Solution(object):
    def postorder(self, root):
        """
        :type root: Node
        :rtype: List[int]
        """
        if not root:
            return []
        if not root.children:
            return [root.val]
        result = []
        for child in root.children:
            result += self.postorder(child)
        result += [root.val]
        return result