python解决自守数问题

注意事项：注意格式

参考代码:

import math


def Check(n):
    t = n
    count = 0
    # 计算n有多少位
    while t != 0:
        t //= 10
        count += 1
    c = int(math.pow(10, count))
    # 通过对比n平方的后count位，与原数进行比较
    b = (n ** 2) % c
    if b == n:
        return 1
    else:
        return 0


a = 200000

print('0', end='  ')
for i in range(1, a):
    if Check(i):
        print(i, end='  ')