自守数

https://www.nowcoder.com/practice/88ddd31618f04514ae3a689e83f3ab8e?tpId=37&&tqId=21322&rp=1&ru=/ta/huawei&qru=/ta/huawei/question-ranking

def cntNums(n):
    cnt = 0
    for i in range(n + 1):
        val = pow(i, 2)
        if int(str(val)[-len(str(i)) : ]) == i:
            cnt += 1
    return cnt
while True:
    try:
        n = int(input())
        print(cntNums(n))
    except:
        break

while True:
    try:
        N=int(input())
        num=0
        for i in range(N):
            if str(i*i).endswith(str(i)):
                num+=1
        print(num)
    except:
        break