计算大整数根

尽管 Python 本身支持大整数,但是在 Python 中使用非常大数字的第 n 个根可能会失败。

x = 2 ** 100
cube = x ** 3
root = cube ** (1.0 / 3)

OverflowError:long int 太大而无法转换为 float

处理这样大的整数时,你需要使用自定义函数来计算数字的第 n 个根。

def nth_root(x, n):
    # Start with some reasonable bounds around the nth root.
    upper_bound = 1
    while upper_bound ** n <= x:
        upper_bound *= 2
    lower_bound = upper_bound // 2
    # Keep searching for a better result as long as the bounds make sense.
    while lower_bound < upper_bound:
        mid = (lower_bound + upper_bound) // 2
        mid_nth = mid ** n
        if lower_bound < mid and mid_nth < x:
            lower_bound = mid
        elif upper_bound > mid and mid_nth > x:
            upper_bound = mid
        else:
            # Found perfect nth root.
            return mid
    return mid + 1

x = 2 ** 100
cube = x ** 3
root = nth_root(cube, 3)
x == root
# True