문제

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

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

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

풀이

아이디어
1. 입력 수가 1000 이하 자연수 이므로, 십자리 수 이하는 항상 등차수열
2. 등차수열의 세 요소는 평균 값과 두번째 수의 값이 동일

코드

N = int(input())

def checkSameDiff(num):
    numLen = len(str(num))
    #두 자리 이하는 항상 등차수열
    if numLen <= 2:
        return True
    else:
        #이어지는 세 수의 평균 값이 중간 값과 동일
        avg = sum(map(int,str(num)))/3
        midNum = (num % (10**2)) // 10
        return True if avg == midNum else False

def countSameDiff(num):
    cnt = 0
    for i in range(num):
        if checkSameDiff(i+1):
            cnt += 1
    return cnt

print(countSameDiff(N))

검토

• 문제 이해에 어려움이 있었다. 함수를 잘 짜 두고도 제출 시 실패로 떨어져서 문제를 제차 확인해보니, 첫 줄에 입력이 있고 결과값 출력이 요구되는 문제였다.

출처 : https://www.acmicpc.net