653. 两数之和 IV - 输入 BST

给定一个二叉搜索树和一个目标结果,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。

案例 1:
输入:
5
/
3 6
/ \
2 4 7

Target = 9

输出: True

案例 2:
输入:
5
/
3 6
/ \
2 4 7

Target = 28
输出: False

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def findTarget(self, root: TreeNode, k: int) -> bool:
        res =[]
        
        def Get(root):
            if root is None:
                return []
            return [root.val] + Get(root.left) +Get(root.right)
        
        res = Get(root)
        
        for i,num in enumerate(res):
            if k-num in res and res.index(k-num) != i:
                return True
        
        
        return False

你可能感兴趣的:(653. 两数之和 IV - 输入 BST)