【Leetcode】【python】Binary Tree Preorder Traversal

题目大意

二叉树前序遍历
挑战：迭代解题

解题思路

递归简单
迭代思路：博客

            　　1　　　　　　　/     \　　　　　 2         3　　　 /     \    /    \　　  4       5  6      7 

使用一个栈。步骤为：

一，先遍历节点1，并入栈，如果有左孩子，继续遍历并入栈，一直到栈为{1，2，4}。

二，开始弹栈，当栈顶元素为2时，弹出2，并检测2存在右孩子5，于是遍历5并入栈，此时栈为{1，5}。

三，弹出5，5没有左右孩子，继续弹栈，将1弹出后，栈为{}。

四，由于1存在右孩子，则继续按照以上步骤入栈出栈。{3, 6}->{7}->{}，结束。

栈的状态变化：{1}->{1,2}->{1,2,4}->{1,2}->{1}->{1,5}->{1}->{}->{3}->{3,6}->{3}->{}->{7}->{}。

代码

递归

class Solution(object):    def _preorderTraversal(self, root, result):        if root:            result.append(root.val)            self._preorderTraversal(root.left, result)            self._preorderTraversal(root.right, result)    def preorderTraversal(self, root):        """        :type root: TreeNode        :rtype: List[int]        """        if root == None:            return []        result = []        self._preorderTraversal(root, result)        return result

迭代

class Solution(object):    def preorderTraversal(self, root):        """        :type root: TreeNode        :rtype: List[int]        """        ret = []        stack = [root]        while stack:            node = stack.pop()            if node:                ret.append(node.val)                stack.append(node.right)                stack.append(node.left)        return ret

总结