LeetCode刷题日记2022-3-21/653. 两数之和 IV - 输入 BST-中序遍历&双指针

题目描述

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

示例 1：

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

示例 2：

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

提示:

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

题解思路

• 我们先将搜索二叉树转化为数组形式
• 然后low，high指针分别指向数组的头跟尾
  • 当low+high<k时，low后移
  • 当low+high>k时，high前移

题解代码

class Solution:
    def findTarget(self, root: Optional[TreeNode], k: int) -> bool:
        ans=[]
        def midsearch(node):
            if node.left:
                midsearch(node.left)
            ans.append(node.val)
            if node.right:
                midsearch(node.right)
        midsearch(root)
        low,high=0,len(ans)-1
        while low<high:
            if ans[low]+ans[high]>k:
                high-=1
            elif ans[low]+ans[high]<k:
                low+=1
            else:
                return True
        return False