[백준 1065번] 한수(파이썬)

문제

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

 

입력

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

출력

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

 

예제 입력

예제 1
110

예제 2
1

예제 3
210

예제 4
1000

예제 출력

출력 1
99

출력 2
1

출력 3
105

출력 4
144

 

---

 

풀이

등차수열 : 연속된 숫자의 차이가 일정
어떤 숫자가 주어졌을 때 그 숫자의 각 자리 수가 등차수열이 만족해야 합니다.

예를들면 123 숫자가 있으면 각 자리를 1, 2, 3 으로 나누면  각 자리는 1씩 차이가 남으로 등차수열이 만족합니다.
100 > 1, 0, 0 (등차수열 만족 X)
1 ~ 99는 전부 등차수열을 만족합니다.

코드를 순서대로 작성한다면
1. 함수 생성
2. 함수에 정답을 리턴할 변수를 선언
3. 정답 변수에 1 ~ 99를 카운트 +1
4. 100부터는 등차수열이 만족하는지 확인 후 카운트 +1
5. 정수를 입력 받고, 입력을 함수에 대입 후 출력

---

작성 코드

def hansu(n):
    answer = 0
    
    for i in range(1, n+1): # 1 ~ n 까지 반복
        A = list(map(int, str(i))) # 각 자리수를 나눔
        
        if i < 100: # 99까지는 +1
            answer += 1
        elif A[0] - A[1] == A[1] - A[2]: # 각 자리수르 차이가 같으면 등차수열 성립 카운트 +1
            answer += 1
	return answer
    
N = int(input())
print(hansu(N)) # 입력된 정수를 함수에 대입 후 출력