[백준] 1065번 > 한수

www.acmicpc.net/problem/1065

1065번: 한수

문제

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

입력

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

출력

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

예제 입력 1

110

예제 출력 1

99

예제 입력 2

1

예제 출력 2

1

예제 입력 3

210

예제 출력 3

105

예제 입력 4

1000

예제 출력 4

144

 

 

---

풀이

등차수열을 계산해야 하는 문제이다. 나의 논리는 다음과 같다.

한자릿수, 두자릿수는 무조건 한수이다. 우리가 계산해야 할 것은 세 자릿수 N의 범위가 1000까지 이므로 완전 탐색으로 구현하였다.

N의 백의 자리수와 십의 자릿수를 뺀 것과 십의 자릿수와 일의 자릿수가 서로 같다면 등차수열을 이루고 있으므로 한수이다. 

n = int(input())
answer = 0
for i in range(1,n+1):
    temp = str(i)
    if len(temp) == 1 or len(temp) == 2:
        answer += 1
    elif len(temp) == 3:
        if (int(temp[0]) - int(temp[1])) == (int(temp[1]) - int(temp[2])):
            answer += 1
print(answer)