leetcode_653. 两数之和 IV - 输入 BST（C++，中序遍历+二分查找）

最新推荐文章于 2023-02-25 22:41:55 发布

Regeon Chen

最新推荐文章于 2023-02-25 22:41:55 发布

阅读量217

点赞数 1

分类专栏： C++_题解文章标签：数据结构算法 leetcode c++ 二叉树

本文链接：https://blog.youkuaiyun.com/qq_45698847/article/details/105925032

版权

C++_题解专栏收录该内容

40 篇文章

订阅专栏

题目

原题链接

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

案例 1:

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

Target = 9

输出: True

案例 2:

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

Target = 28

输出: False

思路

首先，利用“二叉搜索树的中序遍历为升序序列”这个特性，将二叉搜索树的所有节点按升序存入数组。
接下来的任务就是该如何找到两个和为 k 的数了。这里我们用双指针，初始分别指向数组一开始与数组末尾，做一个判断：l+r > k ：则 r-- ；l+r < k ：则l++
二分的边界条件是 l>=r

/**
 * 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:
    vector<int> ans;
    bool findTarget(TreeNode* root, int k) {
        inorder(root);
        int l = 0, r = ans.size() - 1;
        while(l < r) { //二分查找
            if(ans[l] + ans[r] == k) return true;
            else if(ans[l] + ans[r] < k) l ++;
            else r --;
        }
        return false;
    }
    void inorder(TreeNode* root) { //中序遍历
        if(root) {
            inorder(root -> left);
            ans.push_back(root -> val);
            inorder(root -> right);
        }
    }
};