from collections import Counter
class TreeNode:
   def __init__（self, data，left = None，right = None）：
      self.val = data
      self.left = left
      self.right = right

class Solution:
   def solve（self，root，target）：
      count = Counter（[0]）
      ans = prefix = 0

      def dfs（node）：
         nonlocal ans，prefix
         if node：
            prefix + = node.val
            ans + = count[prefix-target]
            count[prefix] + = 1
            dfs（node.left）
            dfs（node.right）

            count[prefix] -= 1
            prefix -= node.val
      dfs（root）
      返回答案
     
ob = Solution()
root = TreeNode（3）
root.left = TreeNode（2）
root.right = TreeNode（4）
root.right.left = TreeNode（1）
root.right.left.right = TreeNode（2）
k = 5
print（ob.solve（root，k））

root = TreeNode(3)
root.left = TreeNode(2)
root.right = TreeNode(4)
root.right.left = TreeNode(1)
root.right.left.right = TreeNode(2)
5

2