题目描述:https://www.lintcode.com/problem/closest-binary-search-tree-value/description
看到这个题首先想到的是利用二叉搜索树的特点递归搜索。目标值比当前值大就向左查找,小就向右查找。然后比较当前值和向左向右找到的值哪个更接近目标。
class Solution {
public:
int closestValue(TreeNode * root, double target) {
if (root->val - target == 0) return root->val;
if (target>root->val&&root->right != NULL)
{
int next = closestValue(root->right, target);
if (abs(next - target)<abs(root->val - target)) return next;
else return root->val;
}
else if (target<root->val&&root->left != NULL)
{
int next = closestValue(root->left, target);
if (abs(next - target)<abs(root->val - target)) return next;
else return root->val;
}
return root->val;
}
};
但是仔细想一下,这个题是不需要"回溯"这个步骤的,我们,因为可能存在最接近值的路径只有一条,不存在尝试了一条之后返回父节点再次寻找下一条路径的步骤,所以只要向左向右查找到叶子就一定可以枚举到所有可能的节点。
class Solution {
public:
int closestValue(TreeNode* root, double target) {
int res = root->val;
while (root) {
if (abs(res - target) >= abs(root->val - target)) {
res = root->val;
}
root = target < root->val ? root->left : root->right;
}
return res;
}
};