【牛客网华为机试】HJ99 自守数

题目

描述

自守数是指一个数的平方的尾数等于该数自身的自然数。例如：25^2 = 625，76^2 = 5776，9376^2 = 87909376。请求出n(包括n)以内的自守数的个数

本题有多组输入数据，请使用while(cin>>)等方式处理

输入描述：

int型整数

输出描述：

n以内自守数的数量。

示例1

输入：

5
2000

输出：

3
8

解题思路

（1）遍历0到输入数字，对于位数为5，6，0，1的数字进行自守数判断

（2）如果为自守数，则count+1

代码

while True:
    try:
        num = int(input())+1
        count = 0
        for i in range(num):
            if str(i)[-1] in ("5", "6", "0", "1"):
                if str(i**2)[-len(str(i)):]==str(i)[-len