270.Closest Binary Search Tree Value的C++解法

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/musechipin/article/details/85229404

题目描述: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;
    }
};

猜你喜欢

转载自blog.csdn.net/musechipin/article/details/85229404