Python整数root/power对求解练习：代码优化及实现疑问咨询

关于abs确定root范围的合理性 #

你用abs(num_value)作为root范围的做法不合理，这个范围过于宽泛，会产生大量无效循环。比如输入16时，root会遍历-16到15，但实际上只有-4、-2、2、4这几个数的2-5次方可能等于16，其余数的计算都是无用功，完全没必要遍历。

优化建议与实现 #

核心优化方向 #

1. 缩小root搜索范围
   • 正数场景：对于pwr≥2，root的绝对值最大只需到int(num_value ** (1/pwr)) + 1。因为当root的绝对值超过这个值时，root**pwr必然大于num_value，无需再遍历。
   • 负数场景：只有奇数次方能得到负数，因此pwr只需遍历3、5；同时root的绝对值最大到int(abs(num_value) ** (1/pwr)) + 1，原理同正数场景。
2. 提前终止循环：找到符合条件的root和pwr后，立即终止所有循环，避免无用计算。
3. 分场景处理：将正数、负数、0的逻辑分开，减少不必要的遍历（比如负数无需遍历偶次方）。

优化后的代码 #

num_value = int(input("Enter a number value:"))
found = False

if num_value > 0:
    # 处理正数：遍历pwr 2-5，每个pwr对应更小的root范围
    for pwr in range(2, 6):
        current_max_root = int(num_value ** (1 / pwr)) + 1
        for root in range(-current_max_root, current_max_root + 1):
            if root ** pwr == num_value:
                print(f'{num_value} is equivalent to {root} to the power of {pwr}')
                found = True
                break
        if found:
            break
elif num_value < 0:
    # 处理负数：仅遍历奇数次方3、5
    abs_num = abs(num_value)
    for pwr in [3, 5]:
        current_max_root = int(abs_num ** (1 / pwr)) + 1
        for root in range(-current_max_root, current_max_root + 1):
            if root ** pwr == num_value:
                print(f'{num_value} is equivalent to {root} to the power of {pwr}')
                found = True
                break
        if found:
            break
else:
    # 处理0的情况：0的2-5次方都是0，符合要求
    print(f'{num_value} is equivalent to 0 to the power of 2')
    found = True

if not found:
    print('I was not able to find a root/power integer pair for this number.')

优化效果说明 #

• 循环次数大幅减少：比如输入32时，原代码会遍历-32到31共63个root，每个root遍历4个pwr，总计252次计算；优化后仅需36次左右的计算，效率提升明显。
• 逻辑更清晰：分场景处理后，代码可读性更强，也避免了无效的pwr遍历。

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