From 1abe74373db8f97dcbb9227f6f8ce34a1713bbd4 Mon Sep 17 00:00:00 2001 From: Trisha Moitra Date: Fri, 28 Oct 2022 01:19:22 +0530 Subject: [PATCH 1/2] Added code for Binary tree path sum Binary tree path sum #7135 --- bt_path_sum.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 bt_path_sum.py diff --git a/bt_path_sum.py b/bt_path_sum.py new file mode 100644 index 000000000000..7dab68155573 --- /dev/null +++ b/bt_path_sum.py @@ -0,0 +1,10 @@ +def pathSum(self, root: TreeNode, target_sum: int) -> int: + def dfs(node: TreeNode, sums: List[int]) -> int: + if not node: + return 0 + sums = [s + node.val for s in sums] + [node.val] + ans = sums.count(target_sum) + ans += dfs(node.left, sums) if node.left else 0 + ans += dfs(node.right, sums) if node.right else 0 + return ans + return dfs(root, []) From 83cc6ee1d75c3a49823e31bbb0b741ae9587b06b Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 27 Oct 2022 19:52:03 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- bt_path_sum.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/bt_path_sum.py b/bt_path_sum.py index 7dab68155573..7d057c290aea 100644 --- a/bt_path_sum.py +++ b/bt_path_sum.py @@ -1,10 +1,11 @@ def pathSum(self, root: TreeNode, target_sum: int) -> int: - def dfs(node: TreeNode, sums: List[int]) -> int: - if not node: - return 0 - sums = [s + node.val for s in sums] + [node.val] - ans = sums.count(target_sum) - ans += dfs(node.left, sums) if node.left else 0 - ans += dfs(node.right, sums) if node.right else 0 - return ans - return dfs(root, []) + def dfs(node: TreeNode, sums: List[int]) -> int: + if not node: + return 0 + sums = [s + node.val for s in sums] + [node.val] + ans = sums.count(target_sum) + ans += dfs(node.left, sums) if node.left else 0 + ans += dfs(node.right, sums) if node.right else 0 + return ans + + return dfs(root, [])