力扣小白刷题之653题两数之和Ⅳ-输入BST

题目描述

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

思路

参考自：https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst/solution/liang-shu-zhi-he-iv-by-leetcode/

使用中序遍历得到有序数组之后，再利用双指针对数组进行查找。
应该注意到，这一题不能用分别在左右子树两部分来处理这种思想，因为两个待求的节点可能分别在左右子树中。

• 中序遍历BST，得到升序数组list
  

• 使用两个指针 l 和 r 作为 list 的头部索引和尾部索引。然后执行以下操作：

  1. 检查 l 和 r 索引处两元素之和是否等于 k，是的话立即返回 true
  2. 如果当前两元素之和小于 k，则更新 l 指向下一个元素，因为需要增大两数之和时，应该增大较小数。
  3. 如果大于 k，则更新 r 指向上一个元素，因为需要减小两数之和时，应该减小较大数。
  4. 重复 2、3，直至 l >= r
  5. 此时如果还没有找到两元素之和等于 k，则返回false。

• 时间复杂度：O(n)，n 是树中节点的数量，本方法需要中序遍历整棵树。

• 空间复杂度：O(n)，list 中存储 n 个元素。

代码