leetcode 653. 两数之和 IV - 输入 BST(二叉搜索树)

本文介绍了一种在二叉搜索树中寻找两数之和等于特定目标值的有效算法。通过递归前序遍历结合二叉搜索树特性进行查找,避免了重复和无效搜索,提高了效率。

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

题目描述:

 

给定一个二叉搜索树和一个目标结果,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。

案例 1:

输入: 
    5
   / \
  3   6
 / \   \
2   4   7

Target = 9

输出: True

 

案例 2:

输入: 
    5
   / \
  3   6
 / \   \
2   4   7

Target = 28

输出: False

思路:遍历二叉搜索树每一个结点(我的代码采用的是前序遍历),遍历每个结点时求出sum减去当前结点值的差,再用二叉搜索树的方式查找是否含有这个差,如果存在则返回true,否则返回false。注意:排除结点值是sum值一半的情况。

看了网上其他大神的代码:貌似也可以建一个哈希表。(下次尝试)

代码:

class Solution {
public:
    bool findTarget(TreeNode* root, int k) {
        if(root==NULL||(!root->left&&!root->right))return false;
        int flag=0;
        root=preorder(root,root,k,flag);
        return flag;
    }
    TreeNode * preorder(TreeNode * cur,TreeNode *root,int k,int& flag){
        if(cur==NULL)return NULL;
        if(flag==1)return cur;
        if(cur->val!=k-cur->val)//例如:搜索和为6,假如二叉树有3,搜索时搜索的是3(因为6-3=3),会出错,所以要排除这种情况
            flag=search(root,k-cur->val);
        cur->left=preorder(cur->left,root,k,flag);
        cur->right=preorder(cur->right,root,k,flag);
        return cur;
        }
    bool search(TreeNode * root,int k){
        if(root==NULL)return false;
        if(root->val==k)return true;
        if(root->val<k)return search(root->right,k);
        if(root->val>k)return search(root->left,k);
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值