LeetCode刷题笔记589:N叉树的前序遍历（Python实现）

原创于 2019-06-03 14:49:29 发布 · 212 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #Python #算法

LeetCode 专栏收录该内容

50 篇文章

订阅专栏

本文介绍了一种数据结构——N叉树的前序遍历算法，包括递归和非递归两种实现方式。递归方法简洁明了，而非递归方法则使用栈来辅助完成遍历过程。

题目描述：

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

例如，给定一个 3叉树 :

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

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

Solution：

1.递归实现：

Code:

#递归
class Solution:
    def preorder(self,root):
        '''
        
        :param root:Node 
        :return: list[int]
        '''
        res = []
        def helper(root):
            if not root:
                return
            res.append(root.value)
            for child in root.children:
                helper(child)
        helper(root)
        return res

2.非递归：使用了栈

#非递归
    def preorder2(self,root):
        '''
        
        :param root: Node
        :return: List[int]
        '''
        if not root:
            return
        s = [root]
        res = []
        while len(s) > 0:
            node = s.pop()
            res.append(node.val)
            s.extend(node.children[::-1])
        return res