這是一個 leet 代碼問題,因此沒有定義樹。但這里是 TreeNode 類定義:(不適用于我的問題,但無論如何都在這里):Definition for a binary tree node.class TreeNode:def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right所以,這就是我所做的,但失敗了,因為列表為空時無法訪問。問題出現在### 指示的行上。class Solution:def postorderTraversal(self, root: TreeNode) -> List[int]: stack=[] output=[] node = root while True: if node: if (node.right != None): stack.append(node.right) stack.append(node) node=node.left elif stack: check=stack.pop() if (check.right != None) & (check.right == stack[-1]): ### node=stack.pop() stack.append(check) else: output.append(check.val) node=None else: break return output錯誤:IndexError: list index out of rangeif (check.right != None) & (check.right == stack[-1]):Line 27 in postorderTraversal (Solution.py)ret = Solution().postorderTraversal(param_1)Line 57 in _driver (Solution.py)_driver()Line 68 in <module> (Solution.py)因此,我在 # 指示的行上添加了一個方法來執行堆棧 [-1] 的檢查以避免超出范圍的錯誤。def check_stack(stack):if stack: return stack[-1]else: return Noneclass Solution:def postorderTraversal(self, root: TreeNode) -> List[int]: stack=[] output=[] node = root while True: if node: if (node.right != None): stack.append(node.right) stack.append(node) node=node.left elif stack: check=stack.pop() if (check.right != None) & (check.right == check_stack(stack)): # node=stack.pop() stack.append(check) else: output.append(check.val) node=None else: break return output在這種情況下,是否有更簡單的方法(無需創建新方法和更少的代碼行)來執行堆棧 [-1] 檢查而不影響其他條件語句的順序?
1 回答

白板的微信
TA貢獻1883條經驗 獲得超3個贊
check_stack您可以通過使用切片獲取堆棧的最后一個元素而不是訪問特定索引來避免該函數。
在這種情況下,如果堆棧中沒有元素,您將得到空列表而不是None。
例如:
stack = [1,2]
stack[-1:] # 2
stack = []
stack[-1:] # []
- 1 回答
- 0 關注
- 113 瀏覽
添加回答
舉報
0/150
提交
取消