二叉树的前序遍历

题目描述:给出一棵二叉树,返回其节点值的前序遍历。要求不能用递归。

什么是前序遍历应该已经很清楚了,样例我就省略了。


关于二叉树的前序遍历,我之前已经详细讲过,当时是用的最简单的递归解决这个问题的,详见:点击打开链接,但是递归是有它的天然缺陷的,那就是效率低。所以,有人试图用一种更高效的办法替代递归。就这个问题来说,由于二叉树的前序遍历用的是“深搜”的策略(中序遍历,后序遍历也都是“深搜”),而“深搜”其实一般情况下可以用栈来实现。

回忆一下,前序遍历就是一个“根-左-右”的过程


拿上面这幅图来说,前序遍历的结果应该是:ABDECFG,对于到达的每一个节点都按照先扫描这个节点本身,再扫描左孩子,最后右孩子的过程。所以,我不妨建立这样一个栈--stack

1. stack存储当前节点,一开始,当然是根节点A,stack = [A]

2. 删除stack的栈顶元素,将栈顶元素所存储的值加入到最后我们要的结果列表里面

3. 如果被删除的元素有右孩子,将右孩子加入stack,stack = [C]

4. 如果被删除的元素有左孩子,将左孩子加入stack,stack = [C,B]

可以看出,由于右孩子先加入,所以再把这个新的栈从第一步开始迭代时,就会先处理左孩子(图中的B),以此类推,处理左孩子时,又会先删除该节点(B),再加入该节点(B)的右孩子:stack = [C,E],最后加入左孩子:stack = [C,E,D]...

很明显,我在这里通过栈的特殊结构实现了深搜的策略

看代码吧:

"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None
"""


class Solution:
    """
    @param root: The root of binary tree.
    @return: Preorder in ArrayList which contains node values.
    """
    def preorderTraversal(self, root):
        result = []
        stack = []
        # 也可写成 if root:
        if root != None:
            stack.append(root)
        while len(stack) != 0:
            cur = stack.pop()
            result.append(cur.val)
            if cur.right != None:
                stack.append(cur.right)
            if cur.left != None:
                stack.append(cur.left)
        return result
        # write your code here


需要注意的是第19行的简洁写法可以是if root:,24,26行也都该这样写,但是我现在这样,更容易让大家理解。


  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值