刷題系列 - Python用非遞迴實現二叉樹中序遍歷

刷題沒事會上癮，繼續做了非遞迴中序遍歷。

中序遍歷其實和就是先找到最左邊節點，然後是其上級節點，再到上級節點的右邊節點。

比如下面的中序遍歷結果就是 DBEAFC

非遞迴實現邏輯，我想的這個比較笨。就是用一個佇列做棧，先按照左邊遍歷壓入棧中；當到左邊葉子節點時候，讀取並刪除關聯；推出棧回到上一級節點，如果上級節點沒有右節點，則讀取繼續刪除；如果有，則遍歷右節點；為了防止右邊遍歷返回時候再次讀取父節點；要記錄下上次被推出節點，如果是右節點，則不讀取父節點資訊。

程式碼寫的很難看，不去雕琢了，見笑。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def inorderTraversal(self, root: TreeNode) -> List[int]:
        traversalList = []
        nodeList = []  
        # similar as Preorder traversal, the only change is that the value of node is recored when the node doesn't have left sub-node; new object removedNode as popped node, if a node's right sub-node is removedNode, then it should be popped both.
        if root != None:
            nodeList.append(root)
            currentNode = root
            removedNode = None
            while nodeList != []:
                if currentNode.left != None:
                    currentNode = currentNode.left
                    nodeList.append(currentNode)
                elif currentNode.right == None or currentNode.right == removedNode:
                    if currentNode.right == None:
                        traversalList.append(currentNode.val)
                    removedNode = nodeList.pop()
                    if nodeList!= []:
                        currentNode = nodeList[-1]
                        currentNode.left = None
                elif currentNode.right !=None:
                    traversalList.append(currentNode.val)
                    currentNode = currentNode.right
                    nodeList.append(currentNode)
                        
        return traversalList

來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/22259926/viewspace-2673709/，如需轉載，請註明出處，否則將追究法律責任。