leetcode 653. 两数之和 IV - 输入 BST

最新推荐文章于 2022-05-02 14:37:34 发布

金戈_旭日东升

最新推荐文章于 2022-05-02 14:37:34 发布

阅读量143

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/fang156239305/article/details/106007454

版权

leetcode 专栏收录该内容

223 篇文章 0 订阅

订阅专栏

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

class Solution:
    def findTarget(self, root: TreeNode, k: int) -> bool:
        def inorder(root):
            if not root:
                return []
            return inorder(root.left)+[root.val]+inorder(root.right)
        number = inorder(root)
        n = len(number)

        i =0
        j = n-1
        while i<j:
            if number[i]+number[j]>k:
                j-=1
            elif number[i] + number[j]<k:
                i+=1
            else:
                return True
        return False