(파이썬)백준 1065. 한수

문제

어떤 양의 정수 X의 각 자리가 등차수열을 이룬다면, 그 수를 한수라고 한다. 등차수열은 연속된 두 개의 수의 차이가 일정한 수열을 말한다. N이 주어졌을 때, 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 1,000보다 작거나 같은 자연수 N이 주어진다.

출력

첫째 줄에 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력한다.

구상

1. n을 입력받는다.
2. 1부터 99까지는 모두 한수이므로 이 경우 res에 1을 더한다.
3. 100부터 999까지는 각 자리를 따로 저장한다. a에는 백의 자리, b에는 십의 자리, c에는 일의 자리를 저장하고, a-b가 b-c가 같으면, 즉 등차수열이면 True를 반환한다.
4. 3번을 수행하는 함수 han이 True면 res에 1을 더한다.
5. 2~4 과정을 n회 반복한다.

코드

import sys

def han(n):
    a = n // 100        #백의 자리
    b = n % 100 // 10   #십의 자리
    c = n % 10          #일의 자리

    if (a - b) == (b - c):
        return True
    else:
        return False

n = int(sys.stdin.readline())
res = 0

for i in range(1,n+1):
    if 0 < i < 100:
        res += 1
    else:
        if han(i):
            res += 1

print(res)