Python 自守数求解 - python

@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

源码：

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))