Python實現二叉樹的非遞歸中序遍歷


思路:

1. 使用一個棧保存結點(列表實現);

2. 如果結點存在,入棧,然后將當前指針指向左子樹,直到為空;

3. 當前結點不存在,則出棧棧頂元素,並把當前指針指向棧頂元素的右子樹;

4. 棧不為空,循環2、3部。

代碼如下,解決了leetcode94. Binary Tree Inorder 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 inorderTraversal(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        ret = []
        stack = []
        while root or stack:
            while root:
                stack.append(root)
                root = root.left
            if stack:
                t = stack.pop()
                ret.append(t.val)
                root = t.right
        return ret


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM