浙大 | PTA 習題4-6 水仙花數 (20分)

水仙花數是指一個N位正整數（N≥3），它的每個位上的數字的N次幂之和等于它本身。例如： 153 = 1 3 + 2 3 + 3 3 153 = 1^3 + 2 ^3 + 3^3 153=13+23+33

本題要求編寫程式,計算所有N位水仙花數。

輸入格式:

輸入在一行中給出一個正整數N（3≤N≤7）。

輸出格式:

按遞增順序輸出所有N位水仙花數，每個數字占一行。

輸入樣例:

輸出樣例:

153
370
371
407
           

#include <stdio.h>
//如果用math.h頭檔案， 最後一個testcase會逾時，是以手寫了一個pow的函數來用
int pow_(int n, int p);
int main(void)
{
    int n, start, end;
    scanf("%d", &n);
    start = pow_(10, n - 1);
    end = pow_(10, n );
    
    for (; start < end; start++)
    {
        int temp = start;
        int sum = 0;
        for (int i = 0; i < n; i++)
        {
            sum += pow_(temp % 10, n);
            temp /= 10;
        }
        if (sum == start) printf("%d\n", sum);
    }

    return 0;
}
int pow_ (int n, int p)
{
  	int result = 1;
  	for (int i = 1 ; i <= p;i++)
      	result *= n;
    return result;
}