백준 알고리즘 1065번 한수(python)

2자리 이하의 수는 전부 한수이므로 입력 그대로 출력해주면 된다.

3자리 이상의 수부터는 100부터 끝까지 한수가 몇 개 있는지 확인하고 마지막에 99를 더해주면 된다. 

n = int(input())
if n < 100:
    print(n)
else:
    cnt = 0
    for i in range(100, n+1):
        st = str(i)
        d = int(st[0]) - int(st[1])
        ok = 1
        for j in range(1, len(st)-1):
            if int(st[j]) - int(st[j+1]) != d:
                ok = 0
                break
        if ok:
            cnt += 1
    print(cnt+99)