Leetcode笔记 每日一题 653. 两数之和 IV - 输入 BST (22.03.21)

Leetcode笔记 每日一题 653. 两数之和 IV - 输入 BST (22.03.21)

653. 两数之和 IV - 输入 BST
新手刚开始接触写题解，就当自己学习的笔记，记录一下每日刷题，写得不好多多包涵，也请指出

题目

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

示例一：

输入: root = [5,3,6,2,4,null,7], k = 9
输出: true

示例二：

输入: root = [5,3,6,2,4,null,7], k = 28
输出: false

提示:

• 二叉树的节点个数的范围是 [1, 104].
• 104 <= Node.val <= 104
• root 为二叉搜索树
• -105 <= k <= 105

思路

1.BST为二叉搜索树，由示例可以知道此BTS为升序排列，因此可以利用中序遍历得到一个升序序列arr
2.第二步定义两个左右指针left和right，当left < right：

• 如果arr[left]+arr[right] < k: left+=1
• 如果arr[left]+arr[right] > k: right-=1
• 否则返回True

Python代码

class Solution:
    def findTarget(self, root: TreeNode, k: int) -> bool:
        arr = []
        def inorderBST(node):
            if not node:
                return []
            else:
                inorderBST(node.left)
                arr.append(node.val)
                inorderBST(node.right)
        inorderBST(root)# 调用函数inorderBST(root)
        # 定义左右指针left和right
        left= 0
        right = len(arr)-1
        while left<right:
            if arr[left]+arr[right] < k:
                left += 1
            elif arr[left]+arr[right] > k:
                right -= 1
            else:
                return True
        return False

【参考】官方三种题解方法 ：分别是哈希表+深度、哈希表+广度以及双指针