字符串结尾判断 | HJ99 自守数

#最优解
while True:
    try:
        n = int(input())
        num =0
        for i in range(n):
            b = str(i**2)
            if b.endswith(str(i)):
                num+=1
        print(num)
    except:
        break

# 我的代码
while True:
    try:
        n = int(input())
        cnt = 0
        for i in range(n+1):
            i_len = len(str(i))
            if str(i**2)[i_len*(-1):] == str(i):
                cnt += 1
        print(cnt)

    except:
        break