python求解自守数

题目要求：

数学的世界有很多很漂亮的数字或者数字集合，比如梅森数，哥德巴赫猜想，水仙花数，完全数，自守数。今天我们就一起来探究一下自守数。自守数是指一个正整数的平方的尾数等于该数自身的自然数。例如：25^2 = 625，76^2 = 5776，9376^2 = 87909376。请统计出n以内（包括n）的自守数的个数。

输入格式:

第一行输入n,接下去n行随后输入n个正整数Data。

输出格式:

在一行中输出n个 Data以内自守数的个数，所有输出在一行内完成，末尾没有多余空格。。

输入样例:

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

5
1
2
3
4
5

输出样例:

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

2 2 2 2 3

代码实现1:

ss = int(input())
ls = []
for i in range(ss):
    ls.append(int(input()))
bb = []
for x in ls:
    t = 1
    for i in range(1, x+1):
        k = len(str(i))
        n = i * i % (10 ** k)
        if n == i:
            t += 1
    bb.append(t)
bb = [str(x) for x in bb]
print(" ".join(bb))

代码实现2:

def ZsData(k):
    if k == int(str(k * k)[-len(str(k)):]):
        return True
    else:
        return False


n = int(input())
ls = []
for i in range(n):
    ls.append(int(input()))

Ncount=[]
for num in ls:
    count = 0
    for k in range(0, num + 1):
        if ZsData(k):
            count += 1
    Ncount.append(count)


for i in range(len(Ncount)):
        print(Ncount[i], end=' ' if i != len(Ncount)-1 else '')