DFS+双栈常规题,题目如下
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/
4 8
/ /
11 13 4
/
7 2 1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。
题解
首先分析复杂度,这题我是拿DFS去实现,因为需要遍历到的其实就是最后的叶子节点,这个时候肯定就是选择DFS了,然后使用两个栈来辅助实现,一个栈用来储存节点,另一个栈用来储存节点此时累加的得到的值
class Solution(object):
def hasPathSum(self, root, sum):
"""
:type root: TreeNode
:type sum: int
:rtype: bool
"""
if not root:
return False
nodeStack = [root]
# numberStack = [1]
valueStack = [root.val]
while nodeStack:
node = nodeStack.pop()
# number = numberStack.pop()
value = valueStack.pop()
if not node.left and not node.right:
if value == sum:
return True
else:
continue
if node.left != None:
nodeStack.append(node.left)
valueStack.append(node.left.val+value)
if node.right != None:
nodeStack.append(node.right)
valueStack.append(node.right.val+value)
return False
执行用时 : 44 ms, 在Path Sum的Python提交中击败了99.65% 的用户
内存消耗 : 15.4 MB, 在Path Sum的Python提交中击败了25.82% 的用户