给定一个二叉搜索树和一个目标结果,如果 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 BST2vec(TreeNode* root, vector<int>& vals){
if(root){
BST2vec(root->left, vals);
vals.push_back(root->val);
BST2vec(root->right, vals);
}
}
bool findTarget(TreeNode* root, int k) {
vector<int> vals;
BST2vec(root, vals);
int i=0, j=vals.size()-1;
while(i<j){
if(vals[i]+vals[j]>k) j--;
else if(vals[i]+vals[j]<k) i++;
else return true;
}
return false;
}
};
中序遍历得到有顺序的一组数后,设置前后两个指针,搜索所求数的加数。