两数之和 IV - 输入 BST(两数之和要联想到hash)

最新推荐文章于 2024-07-21 13:38:21 发布

日就月将

最新推荐文章于 2024-07-21 13:38:21 发布

阅读量524

点赞数

分类专栏：算法文章标签：算法 java leetcode

本文链接：https://blog.csdn.net/weixin_45830214/article/details/123765718

版权

算法专栏收录该内容

30 篇文章 0 订阅

订阅专栏

文章目录

题目
一、解题思路
二、代码
- 1.前序遍历
- 2.深度优先

题目

今天分享一道力扣上的题

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

源自力扣

一、解题思路

一般遇到两数之和类型的题目，通常都会用到hash，记下标的用hashmap，单纯找是否存在的用hashset。

对于一个节点x的值，只需要我们检查哈希表中是否存在k-x即可。如果可以找到，返回true，否则，把当前值添加到哈希表中。
如果遍历完整棵树都没有找到，即返回false

思路有了，接下来的问题就是如何去遍历了。
方法有很多种，比如前序后序中序层序都可以，选择多多，
这里展示深度优先和前序。

二、代码

1.前序遍历

Dqueu实现的是栈的先进后出
前序依次遍历存在(k-root.val)返回true；否则添加到hashset中。

public boolean findTarget(TreeNode root, int k) {
        Set<Integer> s = new HashSet<>();
        Deque<TreeNode> eq = new LinkedList<>();
        while(!eq.isEmpty()||root!=null){
            while(root!=null){
                eq.push(root);
                if(s.contains(k-root.val)){
                    return true;
                }else{
                    s.add(root.val);
                }
                root=root.left;
            }
            root = eq.pop();
            root =root.right;
        }
        return false;
    }

2.深度优先

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