题目
给定一个二叉搜索树 root 和一个目标结果 k,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
示例 1:
输入: root = [5,3,6,2,4,null,7], k = 9
输出: true
示例 2:
输入: root = [5,3,6,2,4,null,7], k = 28
输出: false
提示:
- 二叉树的节点个数的范围是 [1, 104].
- -104 <= Node.val <= 104
- root 为二叉搜索树
- -105 <= k <= 105
答案
哈希表+遍历
链接
解
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> getordered(TreeNode* root) {
vector<int> ans;
if(root != NULL) {
vector<int> tmp;
tmp = getordered(root->left);
ans.insert(ans.end(),tmp.begin(), tmp.end());
ans.push_back(root->val);
tmp = getordered(root->right);
ans.insert(ans.end(),tmp.begin(), tmp.end());
}
return ans;
}
bool findTarget(TreeNode* root, int k) {//双指针
vector<int> ordered = getordered(root);
int left = 0, right = ordered.size() - 1;
while(left < right) {
if(ordered[left] + ordered[right] < k) {
left++;
}
else if(ordered[left] + ordered[right] > k) {
right--;
}
else {
return true;
}
}
return false;
}
};