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
【参考】官方三种题解方法 :分别是哈希表+深度、哈希表+广度以及双指针