题目地址:
https://leetcode.com/problems/preimage-size-of-factorial-zeroes-function/description/
给定一个正整数 k k k,求有多少个非负整数的阶乘末尾恰好有 k k k个 0 0 0。
设 f ( k ) f(k) f(k)是有多少个非负整数的阶乘末尾的 0 0 0的个数小于等于 k k k,那么答案就是 f ( k ) − f ( k − 1 ) f(k)-f(k-1) f(k)−f(k−1)。而一个数的阶乘末尾有多少个 0 0 0,取决于其阶乘里因子 5 5 5的个数,即, x ! x! x!的末尾 0 0 0的个数为 ⌊ x 5 ⌋ + ⌊ x 5 2 ⌋ + . . . \lfloor \frac{x}{5}\rfloor+\lfloor \frac{x}{5^2}\rfloor+... ⌊5x⌋+⌊52x⌋+...。而一个数越大,其阶乘末尾的 0 0 0的个数不会变少的,所以 f ( k ) f(k) f(k)可以二分来做,取右端点的时候要保证右端点的阶乘的 0 0 0的个数是大于 k k k的。代码如下:
class Solution {
public:
int preimageSizeFZF(int k) {
if (!k) return 5;
auto f = [](long x) {
long res = 0;
while (x) res += x / 5, x /= 5;
return res;
};
auto calc = [&f](long k) {
long l = k, r = 5 * (k + 1);
while (l < r) {
long mid = l + (r - l + 1 >> 1);
if (f(mid) <= k) l = mid;
else r = mid - 1;
}
return l;
};
return calc(k) - calc(k - 1);
}
};
时间复杂度 O ( log 2 k ) O(\log^2k) O(log2k),空间 O ( 1 ) O(1) O(1)。