知乎三面的一道题
653. Two Sum IV - Input is a BST
方法一:hash
使用哈希表来保存BST中节点的值。每当我们将新节点的值插入哈希表时,我们都会检查哈希表是否包含k-node.val
class Solution {
public:
bool findTarget(TreeNode* root, int k) {
unordered_set<int> set;
return dfs(root,set,k);
}
bool dfs(TreeNode* root,unordered_set<int> &set, int k){
if(!root) return false;
if(set.count(k-root->val)) return true;
set.insert(root->val);
return dfs(root->left,set,k)||dfs(root->right,set,k);
}
};
方法二: 中序遍历转换为有序数组
通过使用中序遍历来保存BST中节点的值。然后,我们使用两个指针,从数组的开头和结尾开始,找出是否有总和k
class Solution {
public:
bool findTarget(TreeNode* root, int k) {
vector<int> nums;
inorder(root, nums);
for(int i = 0, j = nums.size()-1; i<j;){
if(nums[i] + nums[j] == k)return true;
(nums[i] + nums[j] < k)? i++ : j--;
}
return false;
}
void inorder(TreeNode* root, vector<int>& nums){
if(root == NULL)return;
inorder(root->left, nums);
nums.push_back(root->val);
inorder(root->right, nums);
}
};
方法三: 二分
对于每个节点,我们检查k - node.val
在BST中是否存在。
class Solution{
public:
bool findTarget(TreeNode* root, int k) {
return dfs(root, root, k);
}
bool dfs(TreeNode* root, TreeNode* cur, int k){
if(cur == NULL)return false;
return search(root, cur, k - cur->val) || dfs(root, cur->left, k) || dfs(root, cur->right, k);
}
bool search(TreeNode* root, TreeNode *cur, int value){
if(root == NULL)return false;
return (root->val == value) && (root != cur)
|| (root->val < value) && search(root->right, cur, value)
|| (root->val > value) && search(root->left, cur, value);
}
};