leetcode 653. Two Sum IV - Input is a BST

最新推荐文章于 2020-07-21 11:03:54 发布

沐沐余风

最新推荐文章于 2020-07-21 11:03:54 发布

阅读量949

点赞数

分类专栏： leetcode 文章标签： leetcode BST 二叉搜索树中序遍历

本文链接：https://blog.csdn.net/my_lovely_lemon_tree/article/details/77816823

版权

leetcode 专栏收录该内容

171 篇文章 0 订阅

订阅专栏

1.题目

Given a Binary Search Tree and a target number, return true if there exist two elements in the BST such that their sum is equal to the given target.

翻译：给一棵二叉搜索树和一个目标值，判断是否存在两个节点的值相加等于target

Example 1:
Input:

Target = 9
Output: True

Example 2:
Input:

Target = 28

Output: False

2.分析

要找出任意两个节点，他们的和为target。

由于是BST，用中序遍历，得到的数组为递增数组，从数组两端开始判断i =0,j=len-1
a[i]+a[j] == target 返回 true
a[i]+a[j] < target ++i
a[i]+a[j] > target –j

3.代码

class Solution {
public:
    bool findTarget(TreeNode* root, int k) {
        vector<int> nums;
        stack<TreeNode*> nodes;
        while (!nodes.empty()||root) {
            if (root) {
                nodes.push(root);
                root = root->left;
            }
            else {
                root = nodes.top();
                nodes.pop();
                nums.push_back(root->val);
                root = root->right;
            }
        }
        int left = 0, right = nums.size() - 1;
        while (left < right) {
            int sum = nums[left] + nums[right];
            if (sum == k)
                return true;
            else if (sum < k)
                ++left;
            else
                --right;
        }
        return false;
    }
};