leetcode_653_两数之和 IV - 输入 BST

最新推荐文章于 2022-03-21 11:30:27 发布

原创最新推荐文章于 2022-03-21 11:30:27 发布 · 262 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #c++

本文介绍了一种算法，该算法通过中序遍历二叉搜索树(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

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void BST2vec(TreeNode* root, vector<int>& vals){
        if(root){
            BST2vec(root->left, vals);
            vals.push_back(root->val);
            BST2vec(root->right, vals);
        }
    }
    bool findTarget(TreeNode* root, int k) {
        vector<int> vals;
        BST2vec(root, vals);
        int i=0, j=vals.size()-1;
        while(i<j){
            if(vals[i]+vals[j]>k) j--;
            else if(vals[i]+vals[j]<k) i++;
            else return true;
        }
        return false;
    }
};

中序遍历得到有顺序的一组数后，设置前后两个指针，搜索所求数的加数。