Python實現二叉樹的非遞歸先序遍歷
阿新 • • 發佈:2017-11-20
結果 leetcode logs [] 列表 遍歷 不存在 preorder bin
思路:
1. 使用列表保存結果;
2. 使用棧(列表實現)存儲結點;
3. 當根結點存在,保存結果,根結點入棧;
4. 將根結點指向左子樹;
5. 根結點不存在,棧頂元素出棧,並將根結點指向棧頂元素的右子樹;
6. 重復步驟3-6,直到棧空。
LeetCode: 144. Binary Tree Preorder Traversal
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def preorderTraversal(self, root):
"""
:type root: TreeNode
:rtype: List[int]
"""
ret = []
stack = []
while root or stack:
while root:
ret.append(root.val)
stack.append(root)
root = root.left
if stack:
t = stack.pop()
root = t.right
return ret
Python實現二叉樹的非遞歸先序遍歷