653. 两数之和 IV - 输入 BST
题目:给定一个二叉搜索树和一个目标结果,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
案例 1:
输入:
5
/
3 6
/ \
2 4 7
Target = 9
输出: True
案例 2:
输入:
5
/
3 6
/ \
2 4 7
Target = 28
输出: False
class Solution {
public:
void dfs(TreeNode* root, vector<int>& nums)
{
if(root == nullptr) return;
//题目为二叉搜索树,遍历得到的结果必须是有序的
//所以只能是中序遍历
dfs(root->left, nums);
nums.push_back(root->val);
dfs(root->right, nums);
}
bool findTarget(TreeNode* root, int k) {
vector<int> nums;
dfs(root, nums);
int Left = 0;
int 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;
}
};