Python練習題4.17水仙花數

水仙花數是指一個N位正整數（N≥3），它的每個位上的數字的N次冪之和等於它本身。 例如：153=1×1×1+5×5×5+3×3×3。

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

輸入格式:

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

輸出格式:

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

代碼如下：

#!/usr/bin/python
# -*- coding: utf-8 -*-

def sxh(n):
    result = list()
    if n == 3:
        for i in range(100,1000):
            if (i//100)**3 + ((i%100)//10)**3 + ((i%100)%10)**3 == i:
                result.append(i)
    elif n == 4:
        for i in range(1000,10000):
            if (i//1000)**4 + ((i%1000)//100) **4 + (((i%1000)%100)//10) **4 + (((i%1000)%100)%10) **4== i:
                result.append(i)
    elif n == 5:
        for i in range(10000,100000):
            if (i//10000) **5 + ((i%10000)//1000) **5 + (((i%10000)%1000)//100) **5 + ((((i%10000)%1000)%100)//10) **5 + ((((i%10000)%1000)%100)%10) **5== i:
                result.append(i)
    return result

n = int(input())
re = sxh(n)
for i in range(0,len(re)):
    print(re[i])

這個程序不難，細心點就行，其實測試個3位數的水仙花就行了。

我的這個程序應該優化的空間很大，大家下去自行研究，我有時間了也在看看。

讀書和健身總有一個在路上