力扣每日一题

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

题目描述

给定一个二叉搜索树 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

解题思路

搜索二叉树的性质，根节点比左节点大，比右节点小。中序遍历是一个递增的数组。使用哈希表加树的深度搜索的方法解题。 使用哈希表记录遍历到的节点，在遍历到节点v时，发现哈希表中存在k-v.val，说明存在两个值和为k的节点，返回true.若搜索完所有节点都没有发现这样的节点，返回false

AC代码

class Solution {
    Set<Integer> set = new HashSet<>();
    public boolean findTarget(TreeNode root, int k) {
        if(root == null){
            return false;
        }
        if(set.contains(k - root.val)){
            return true;
        }
        set.add(root.val);
        return findTarget(root.left, k) || findTarget(root.right, k);
    }
}

时间复杂度 O(n)

空间复杂度 O(n)

思考总结

对于二叉树的一类题目，通常使用递归的方法解决。熟悉三种遍历方法，以及非迭代遍历方法。使用深度优先搜索和广度优先搜索的方法进行解题。

解法二 广度优先搜索

 class Solution {
  
    public boolean findTarget(TreeNode root, int k) {
         Set<Integer> set = new HashSet<>();
         Queue<TreeNode> queue = new ArrayDeque<>();
         queue.offer(root);
         while(!queue.isEmpty()){
             int size = queue.size();
             for(int i = 0; i < size; i++){
                TreeNode node = queue.poll();
                if(set.contains(k - node.val)){
                    return true;
                }
                set.add(node.val);
                if(node.left != null){
                    queue.offer(node.left);
                }
                if(node.right != null){
                    queue.offer(node.right);
                }
             }
            
         }
         return false;
    }
}