1. 程式人生 > >270.Closest Binary Search Tree Value的C++解法

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

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