第4章-17 水仙花数（20 分）

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

本题要求编写程序,计算所有N位水仙花数。

输入格式:
输入在一行中给出一个正整数N（3≤N≤5）

输出格式:
按递增顺序输出所有N位水仙花数，每个数字占一行。

输入样例:
在这里给出一组输入。例如：

3

输出样例:
在这里给出相应的输出。例如：

153
370
371
407

题目集全集传送门

import math


def digit(n):
    count = 1
    while n // 10 > 0:
        count += 1
        n //= 10
    return count


def narcissistic(n):
    count = digit(n)
    sum = 0
    num = n
    while n // 10 > 0:
        sum += math.pow(n % 10, count)
        n //= 10
    if sum + math.pow(n, count) == num:
        return 1
    return 0


n = int(input())
for i in range(int(math.pow(10, n - 1)), int(math.pow(10, n)) - 1):
    if narcissistic(i):
        print(i)