剑指 Offer II 056. 二叉搜索树中两个节点之和
给定一个二叉搜索树的 根节点 root 和一个整数 k , 请判断该二叉搜索树中是否存在两个节点它们的值之和等于 k 。假设二叉搜索树中节点的值均唯一。
示例 1:
输入: root = [8,6,10,5,7,9,11], k = 12
输出: true
解释: 节点 5 和节点 7 之和等于 12
示例 2:
输入: root = [8,6,10,5,7,9,11], k = 22
输出: false
解释: 不存在两个节点值之和为 22 的节点
提示:
二叉树的节点个数的范围是 [1, 104].
-104 <= Node.val <= 104
root 为二叉搜索树
-105 <= k <= 105
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/opLdQZ
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
方法一:dfs + 哈希表(空间复杂度O(n))
直接遍历存储节点值找有没有满足条件的两数即可。
方法二:迭代器 + 双指针(空间复杂度O(h)
h 是树的深度。参考第6题的思路和第55题的迭代器:
https://blog.csdn.net/weixin_43942435/article/details/122953885
https://blog.csdn.net/weixin_43942435/article/details/124128125
把二叉搜索树看成数组,创建两个相反迭代器,这样就有了两个指针,参考第6题的思路即可。
题解(Java)
方法一:dfs + 哈希表
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
private Set<Integer> set = new HashSet<>();
public boolean findTarget(TreeNode root, int k) {
if (root == null) {
return false;
}
if (set.contains(k - root.val