Leetcode 653. 两数之和IV - 输入BST

皮卡猴

于 2022-05-02 14:37:34 发布

阅读量324

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode刷题文章标签： leetcode 深度优先算法

本文链接：https://blog.youkuaiyun.com/flying_monkey_1/article/details/124540741

leetcode刷题专栏收录该内容

252 篇文章

订阅专栏

这篇博客介绍了如何解决LeetCode第653题——两数之和IV，给定一个二叉搜索树和一个整数k，判断树中是否存在两个节点，它们的和等于k。解决方案是通过深度优先搜索遍历二叉搜索树，利用其特性构建哈希映射，实时检查目标和是否存在。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Leetcode 653. 两数之和IV - 输入BST

1. 问题描述

在这里插入图片描述

2. 思路

遍历二叉搜索树，将遍历到的结果收集到map中,key为当前节点root.Val
- 判断map[k-root.Val]是否存在，如果存在返回true，不存在返回false

3. 代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */

func findTarget(root *TreeNode, k int) bool {
    set := make(map[int]int, 0)
    return dfs(root, k, set)
}
func dfs(root *TreeNode, k int, set map[int]int) bool {
    // 遍历二叉搜索树，将结果放到map中
    if root == nil {
        return false
    }
    if _, ok := set[k-root.Val]; ok {
        return true
    }
    set[root.Val] = root.Val
    // fmt.Printf("set = %v, k = %v, root.Val=%v\n", set, k-root.Val, root.Val)

    left := dfs(root.Left, k, set)
    right := dfs(root.Right, k, set)
    return left || right
}