Python中如何实现无溢出的大整数任意次幂运算？

Python没有名为ipow的内置整数开根函数，但有纯整数运算、支持任意大整数、不会触发浮点数溢出的整数任意次开根方案。
你碰到溢出的核心原因很简单：math.pow会强制把所有入参转成浮点数计算，浮点数有固定的表示范围上限，碰到2**20000这种超大型整数必然报错，就算用Python自带的**运算符写n ** (1/k)，也会自动走浮点计算逻辑，一样会溢出或者丢精度。

方案1：标准库原生调用（Python 3.11+） #

math模块从3.11版本开始，新增了和math.isqrt（整数平方根）对应的整数任意次根函数math.iroot，全程走原生大整数运算，完全不涉及浮点数转换，没有溢出风险。
该函数入参为非负整数n（待开根的数）、正整数k（根次），返回元组(root, is_exact)：

• root是满足root ** k <= n的最大整数，也就是你要的向下取整的开根结果
• is_exact是布尔值，标记n是不是恰好为root的k次完全幂
  示例代码：

import math
# 计算2^20000的三次方根，无溢出
root, is_perfect_power = math.iroot(2**20000, 3)
print(root)

方案2：低版本兼容实现（全Python3版本可用） #

如果使用3.11以下的Python版本，直接用牛顿迭代法写纯整数运算的开根函数就行，收敛速度极快，哪怕是十万位级别的大整数也能毫秒级返回结果：

def integer_nthroot(n: int, k: int):
    if n < 0:
        raise ValueError("待开根值必须为非负整数")
    if k <= 0:
        raise ValueError("根次必须为正整数")
    if n == 0:
        return (0, True)
    # 位运算快速生成初始迭代值，减少迭代次数
    x = 1 << ((n.bit_length() + k - 1) // k)
    while True:
        y = ((k - 1) * x + n // (x ** (k - 1))) // k
        if y >= x:
            return (x, x ** k == n)
        x = y

调用方式和math.iroot完全一致：

root, is_perfect_power = integer_nthroot(2**20000, 3)
print(root)

补充说明 #

别搞混整数幂和开根的场景：Python内置的全局pow函数本身就支持任意精度的整数幂计算，比如pow(2, 20000)可以直接输出正确的大整数结果，完全不会溢出，只有传入1/3这类浮点指数时才会转浮点数计算触发问题；三参数形式pow(base, exp, mod)是模幂运算，用于快速计算大整数幂取模的结果，和开根场景无关。

内容的提问来源于stack exchange，提问作者LogZ