题目描述:
给定一个二叉搜索树和一个目标结果,如果 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 preOrder(TreeNode* root, vector<int>& nums){
if(root){
preOrder(root->left, nums);
nums.push_back(root->val);
preOrder(root->right, nums);
}
}
bool findTarget(TreeNode* root, int k) {
vector<int> nums;
preOrder(root, nums);
int sz = nums.size();
int l = 0, r = sz-1;
while(l < r){
long long sum_val = nums[l] + nums[r];
if(sum_val == k){
return true;
}else if(sum_val < k){
l++;
}else{
r--;
}
}
return false;
}
};