LeetCode 653. 两数之和 IV - 输入二叉搜索树

最新推荐文章于 2024-07-24 11:56:55 发布

芜湖男童

最新推荐文章于 2024-07-24 11:56:55 发布

阅读量174

点赞数 1

分类专栏： LeetCode 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/qq_41046821/article/details/129274811

版权

LeetCode 专栏收录该内容

73 篇文章 0 订阅

订阅专栏

653. 两数之和 IV - 输入二叉搜索树

难度： $\color{Green}{easy}$

题目描述

给定一个二叉搜索树 $roo t$ 和一个目标结果 $k$ ，如果二叉搜索树中存在两个元素且它们的和等于给定的目标结果，则返回 $t r u e$ 。

示例 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, 10^{4}]$ .
$10^{4} <= Node.val <= 10^{4}$
题目数据保证，输入的 $roo t$ 是一棵有效的二叉搜索树
$10^{5} <= k <= 10^{5}$

算法

(深度优先搜索 + 哈希表)

我们可以使用深度优先搜索的方式遍历整棵树，用哈希表记录遍历过的节点的值。

对于一个值为 x 的节点，我们检查哈希表中是否存在 k−x 即可。如果存在对应的元素，那么我们就可以在该树上找到两个节点的和为 k；否则，我们将 x 放入到哈希表中。

如果遍历完整棵树都不存在对应的元素，那么该树上不存在两个和为 k 的节点。

复杂度分析

时间复杂度： $O (n)$ ，其中 $n$ 为二叉搜索树的大小。我们需要遍历整棵树一次。
空间复杂度 : $O (n)$ ，其中 $n$ 为二叉搜索树的大小。

C++ 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    unordered_set<int> hash;

    bool findTarget(TreeNode* root, int k) {
        if (!root) return false;

        if (hash.count(k - root->val)) return true;
        hash.insert(root->val);

        return findTarget(root->left, k) || findTarget(root->right, k);
    }
};