Python题解 | #自守数#

import sys


while True:
    try:
        n = int(input())
        ans = 1
        for i in range(1, n + 1):
            if i % 10 == 1 or i % 10 == 5 or i % 10 == 6:
                a = str(i)
                b = str(i * i)
                if b[-len(a):] == a:
                    ans += 1
        print(ans)



    except:
        break