[BOJ | Python] 1065번: 한수

https://www.acmicpc.net/problem/1065

 

문제

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

 

입력

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

 

출력

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

 

풀이

def is_hansu(n):
    digits = list(map(int, str(n)))
    return n < 100 or (digits[0] - digits[1] == digits[1] - digits[2])

num = int(input())
print(sum(is_hansu(i) for i in range(1, num + 1)))

 

• str(n)을 사용하여 숫자를 문자열로 변환한 후, map(int, str(n))을 통해 각 자리 숫자를 정수 리스트로 만든다.
  • 예시: n = 123이면 → digits = [1, 2, 3]
• 한 자리 또는 두 자리 숫자(100 미만)는 항상 한수이므로 n < 100이면 True를 반환한다.
• 세 자리 숫자는 각 자리 숫자의 등차수열 조건을 확인해야 한다.
  • 예시: digits = [1, 2, 3] → 1 - 2 == 2 - 3이므로 한수!
  • 예시: digits = [2, 4, 7] → 2 - 4 ≠ 4 - 7이므로 한수가 아님!