这是一个 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 关注
- 78 浏览
添加回答
举报
0/150
提交
取消