270.Closest Binary Search Tree Value的C++解法
阿新 • • 發佈:2018-12-25
題目描述: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; } };