python:二叉树的前序遍历

题目描述：

给定一个二叉树，返回它的前序遍历。

示例：

输入: [1,null,2,3]  
   1
    \
     2
    /
   3 

输出: [1,2,3]

前序遍历：首先访问根节点，然后遍历左子树，最后遍历右子树。 

来源：https://leetcode-cn.com/explore/learn/card/data-structure-binary-tree/2/traverse-a-tree/1/

 

 解题思路：

方法1：

递归

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
        '''如果根节点为空，返回空列表'''
        if not root:
            return []
        '''如果根节点不为空，先取出根节点值，再遍历取出左子树值，最后遍历取出右子树值'''
        return [root.val] + self.preorderTraversal(root.left) + self.preorderTraversal(root.right)

方法2：

迭代

按照一下顺序入栈：先根节点，再右节点，最后左节点

class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
        if not root:
            return []
        stack = [root]
        res = []
        while stack:
            cur = stack.pop()  # pop()方法移除list中的最后一个元素，并返回元素的值
            res.append(cur.val)  # append()添加元素的值到res中，cur.val获取cur的当前值
            if cur.right:
                stack.append(cur.right)
            if cur.left:
                stack.append(cur.left)
        return res