关于完善幂完美数（perfect in power）判断函数的技术求助

Hey John, let's dive into your code for checking "perfect in power" numbers and fix the issues step by step. First, let's recap what a perfect-in-power number is: a number that equals the sum of its digits each raised to some positive integer exponent (like your example 153 = 1³ + 5³ + 3³).

原代码存在的核心问题 #

Let's go through the key problems in your current implementation:

• 无限循环风险：你的for循环条件是sum < n，但有些数字会导致sum永远小于n，比如n=10：不管i多大，1^i + 0^i = 1，永远小于10，循环会一直跑下去，根本停不下来。
• pow函数的精度误差：pow返回的是double类型，整数次方计算可能出现精度丢失。例如pow(7,3)可能返回342.99999999999994，强制转成int就变成342，直接导致sum计算错误，让正确的数被误判。
• 不必要的0次方计算：幂完美数的指数通常是正整数，从i=0开始计算没有意义（非0数字的0次方都是1，0的0次方是未定义行为）。

改进后的代码实现 #

Here's a revised version that fixes these issues, with comments explaining each change:

// 自定义整数幂函数，避免double精度问题
int int_power(int base, int exp, int max_limit) {
    int result = 1;
    for (int j = 0; j < exp; j++) {
        // 提前判断溢出：如果当前结果已经超过max_limit，继续乘会更大，直接返回
        if (result > max_limit / base) {
            return max_limit + 1;
        }
        result *= base;
    }
    return result;
}

int CheckIfPerfectInPower(int n) {
    // 处理特殊情况：0和1是幂完美数（1=1^1）
    if (n <= 1) {
        return 1;
    }

    int digit_count = 0;
    int temp = n;
    // 先计算数字的位数，用来确定指数的合理上限
    while (temp > 0) {
        digit_count++;
        temp /= 10;
    }

    // 指数从1开始，上限设为digit_count + 5（足够覆盖所有可能的幂完美数）
    for (int i = 1; i <= digit_count + 5; i++) {
        int sum = 0;
        int userInput = n;
        while (userInput > 0) {
            int digit = userInput % 10;
            // 用自定义的整数幂计算，避免精度问题
            int digit_power = int_power(digit, i, n);
            // 如果当前位的幂已经超过n，sum肯定会超过n，提前终止计算
            if (digit_power > n - sum) {
                sum = n + 1;
                break;
            }
            sum += digit_power;
            userInput /= 10;
        }
        if (sum == n) {
            return 1;
        }
    }
    return 0;
}

关键改进点说明 #

• 自定义整数幂函数：int_power用整数运算计算幂，还加入了溢出判断，避免计算过大的数导致溢出，同时提前终止无意义的计算。
• 合理的循环终止条件：指数上限设为digit_count + 5，因为对于d位数，当指数超过d+5时，即使所有数字都是9，d*9^i会远大于n，不会存在符合条件的情况，彻底避免无限循环。
• 提前终止优化：在计算每一位的幂时，如果当前累加和加上该位的幂已经超过n，直接标记sum为n+1并跳出循环，减少不必要的计算。
• 特殊情况处理：单独处理0和1，符合幂完美数的定义。

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