Closest Binary Search Tree Value
Given a non-empty binary search tree and a target value, find the value in the BST that is closest to the target.
Note:
- Given target value is a floating point.
- You are guaranteed to have only one unique value in the BST that is closest to the target.
10/17/2015 Update Iteration version
/**
* 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:
int closestValue(TreeNode* root, double target) {
int ans = root->val;
while(root != nullptr){
ans = fabs(target - ans) < fabs(target - root->val)? ans: root->val;
if(target > root->val){
root = root->right;
}
else{
root = root->left;
}
}
return ans;
}
};
Recursive version
/**
* 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:
int closestValue(TreeNode* root, double target) {
int tmp;
if(target > root->val && root->right){
tmp = closestValue(root->right, target);
}
else if(target < root->val && root->left){
tmp = closestValue(root->left, target);
}
else{
return root->val;
}
return abs(tmp - target) < abs(root->val - target)? tmp: root->val;
}
};
/**
* 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:
int closestValue(TreeNode* root, double target) {
int ans = root->val;
BST(root, ans, target);
return ans;
}
void BST(TreeNode * root, int & ans, double target){
if(root == NULL) return ;
if(abs(root->val - target) < abs(ans - target)){
ans = root->val;
}
if(root->val < target){//BUG HERE , root->val not ans
BST(root->right, ans, target);
}
else if (root->val > target){
BST(root->left, ans, target);
}
else{
return ;
}
}
};
