编程,要求输出区间【a,b】之间所有4位数的四位攻瑰数。

img

img

def isRose(n):
    return n == sum(map(lambda x: pow(int(x), 4), str(n)))

def checkNumber(n):
    return -1 if len(n) != 4 or len(n) == 4 and n[0] == '0' else int(n)

a, b = input(">>>").split()

if checkNumber(a) == -1 or checkNumber(b) == -1:
    print("数据不符合要求!")
else:
    result = []
    a, b = int(a), int(b)
    if a > b: a, b = b, a
    for i in range(a, b + 1):
        if isRose(i):
            result.append(i)
    
    print("a={} \nb={}".format(a, b))
    if result == []:
        print("此区段没有玫瑰数!")
    else:
        print("四位玫瑰数:", result)