[BOJ/백준] 1065번 한수 | 함수

문제

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

 

입력

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

 

출력

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

 

 

문제 풀이

def x(n):
    if n <= 99:
        return print(n)
    else:
        count = 99
        for i in range(100,n+1):
            num = []
            l = list(map(int, str(i)))
            for j in range(len(l)-1):
                num.append(l[j+1] - l[j])
            if len(set(num)) == 1:
                count += 1
        return print(count)

n = int(input())
x(n)

한수 개념

• 입력값이 한 자리인 경우 길이가 1인 등차수열이다. = 한수
• 입력값이 두 자리인 경우 길이가 2이고 공차가 양수 또는 음수인 등차수열이다. = 한수
• 입력값이 세 자리 이상인 경우 자릿수의 차를 계산해 판별한다.

 

입력값은 정수형 변수 n으로 받았다. 

 

한수의 개수를 출력해야 하기 때문에 n이 99이하인 경우 n을 반환했다.

 

100이상인 경우 반복문을 사용해 100부터 n까지의 숫자를 확인했다.

- 한수의 갯수를 구하는 변수를 count로 만들었다(99부터 시작).

- 숫자의 각 자릿수를 리스트 l로 정의했다.

- append 함수를 사용해 l 요소들의 차를 빈 리스트 num에 추가했다. 

- 집합(set) 자료형에는 중복이 없다는 성질을 사용해 len(set(num))이 1인 경우 count에 1을 더했다.

 

---

 

1065번: 한수