计算n阶乘中尾部零的个数--O(logN)算法

正文

---

问题

设计一个算法，计算出n阶乘中尾部零的个数

样例
11! = 39916800，因此应该返回 2

挑战
O(logN)的时间复杂度

思路

n的阶乘即从1连乘到n的积，尾部0的个数只会由因数2与5的乘积产生，对任意i*2^j都会有小于该数的i*5^j，故因数2的个数>=因素5的个数，所以只需统计因数5^j的个数即可。如果直接计算因数5^2的个数，则算法复杂度较高，那采用什么方法可以降低算法复杂度呢？如下：

对n除5整除，得出因数5的个数；再除5，得出因数25的个数；……；然后累加。
这样i*5^j这个数即会被累加j次，最终1…n中所有5^j次方因数都被累计出来，即可得到最终结果。

代码

def zero(n):
    ret = 0
    while(n):
        n = int(n / 5)
        ret += n
    return ret

算法复杂度很容易分析出来为O(logN)。

找个比较大的数如1000，在IDLE中验证如下：

>>> import math
>>> import re
>>> ret = str(math.factorial(1000))
>>> ret = ret[::-1] # 逆置字符串
>>> ret
'000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000274357720079...'
>>> pat = re.compile(r'[1-9]')
>>> pos = re.search(pat, ret)
>>> pos.start() # 查找首个非0数字的位置，前面的都是0
249
>>> def zero(n):
    ret = 0
    while(n):
        n = int(n / 5)
        ret += n
    return ret

>>> zero(1000)
249