题目地址:
https://leetcode.com/problems/closest-divisors/
给定一个正整数 x x x,要求找到 a , b a,b a,b满足 a b = x + 1 ∨ a b = x + 2 ab=x+1\lor ab=x+2 ab=x+1∨ab=x+2,并且使得 ∣ a − b ∣ |a-b| ∣a−b∣尽可能小。
可以证明解是唯一的。不妨设 a 1 ≤ b 1 , a 2 ≤ b 2 a_1\le b_1,a_2\le b_2 a1≤b1,a2≤b2如果 a 1 b 1 = x + 1 , a 2 b 2 = x + 2 a_1b_1=x+1,a_2b_2=x+2 a1b1=x+1,a2b2=x+2并且 b 1 − a 1 = b 2 − a 2 b_1-a_1=b_2-a_2 b1−a1=b2−a2,那么 a 2 − a 1 = b 2 − b 1 a_2-a_1=b_2-b_1 a2−a1=b2−b1,则 a 2 b 2 ≥ ( a 1 + 1 ) ( b 1 + 1 ) = a 1 b 1 + a 1 + b 1 + 1 ≥ a 1 b 1 + 3 = x + 4 a_2b_2\ge (a_1+1)(b_1+1)=a_1b_1+a_1+b_1+1\ge a_1b_1+3=x+4 a2b2≥(a1+1)(b1+1)=a1b1+a1+b1+1≥a1b1+3=x+4,矛盾。
接着从 x + 2 → 1 \sqrt {x+2}\to 1 x+2→1遍历,枚举 a a a,如果发现 a ∣ ( x + 1 ) a|(x+1) a∣(x+1)或者 a ∣ ( x + 2 ) a|(x+2) a∣(x+2),则找到了一组解。这个解就是最优解。因为 ( x + 1 , x + 2 ) = 1 (x+1,x+2)=1 (x+1,x+2)=1,所以上面的两个整除式子不能同时成立。如果发现 a 1 b 1 = x + 1 , a 2 b 2 = x + 2 , a 2 < a 1 a_1b_1=x+1,a_2b_2=x+2,a_2<a_1 a1b1=x+1,a2b2=x+2,a2<a1,由于 a 2 b 2 > a 1 b 1 a_2b_2>a_1b_1 a2b2>a1b1所以 b 2 > b 1 b_2>b_1 b2>b1,矛盾;如果发现 a 1 b 1 = x + 2 , a 2 b 2 = x + 1 , a 2 < a 1 a_1b_1=x+2,a_2b_2=x+1,a_2<a_1 a1b1=x+2,a2b2=x+1,a2<a1,如果 b 2 < b 1 b_2<b_1 b2<b1,则 a 1 b 1 = x + 2 ≥ ( a 2 + 1 ) ( b 2 + 1 ) ≥ x + 4 a_1b_1=x+2\ge (a_2+1)(b_2+1)\ge x+4 a1b1=x+2≥(a2+1)(b2+1)≥x+4矛盾。
代码如下:
class Solution {
public:
vector<int> closestDivisors(int num) {
for (int x = sqrt(num + 2); x; x--) {
if ((num + 1) % x == 0) return {x, (num + 1) / x};
if ((num + 2) % x == 0) return {x, (num + 2) / x};
}
return {};
}
};
时间复杂度 O ( x ) O(\sqrt x) O(x),空间 O ( 1 ) O(1) O(1)。