1. 程式人生 > >Python3實現求小於10000的同構數

Python3實現求小於10000的同構數

問題:平方後低位的數恰好等於該數的數是同構數。求小於10000的所有同構數。

編譯環境:Python3 + Spyder

一般的程式碼如下:

for i in range(10000):
   k = str(i * i)                            #平方數
    if(len(k) % 2 == 0):
        m = int(k[(len(k) // 2):len(k)])     #取後位數
        if(m == i):
            print(m,end=" ")
    else:
        m = int(k[((len(k) + 1) // 2) - 1:len(k)])
        if(m == i):
            print(m,end=" ")

使用列表解析後:

from math import ceil
print([i for i in range(1,10000) for n in range(len(str(i * i)),len(str(i * i)) + 1) if i == i * i % (10**(ceil(n / 2))) ])

運算結果:

[1, 5, 6, 25, 76, 376, 625, 9376]