[BAEKJOON]백준 4673번: 셀프 넘버 파이썬

문제

셀프 넘버는 1949년 인도 수학자 D.R. Kaprekar가 이름 붙였다. 양의 정수 n에 대해서 d(n)을 n과 n의 각 자리수를 더하는 함수라고 정의하자. 예를 들어, d(75) = 75+7+5 = 87이다.

양의 정수 n이 주어졌을 때, 이 수를 시작해서 n, d(n), d(d(n)), d(d(d(n))), ...과 같은 무한 수열을 만들 수 있다. 

예를 들어, 33으로 시작한다면 다음 수는 33 + 3 + 3 = 39이고, 그 다음 수는 39 + 3 + 9 = 51, 다음 수는 51 + 5 + 1 = 57이다. 이런식으로 다음과 같은 수열을 만들 수 있다.

33, 39, 51, 57, 69, 84, 96, 111, 114, 120, 123, 129, 141, ...

n을 d(n)의 생성자라고 한다. 위의 수열에서 33은 39의 생성자이고, 39는 51의 생성자, 51은 57의 생성자이다. 생성자가 한 개보다 많은 경우도 있다. 예를 들어, 101은 생성자가 2개(91과 100) 있다. 

생성자가 없는 숫자를 셀프 넘버라고 한다. 100보다 작은 셀프 넘버는 총 13개가 있다. 1, 3, 5, 7, 9, 20, 31, 42, 53, 64, 75, 86, 97

10000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 출력하는 프로그램을 작성하시오.

 

입력

입력은 없다.

 

출력

10,000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 증가하는 순서로 출력한다.

 

코드 

def check(num, a: list):
    for n in str(num):
        num += int(n)
    
    if num <= 10000 and a.count(num) > 0:
        a.remove(num)

if __name__ == '__main__':
    selfNumber = list(range(1,10001))
    
    for i in range(1, 10000):
        check(i, selfNumber)

    for i in selfNumber:
        print(i)

 

 

코드 실행 결과

 

 

코드 설명

1부터 1000까지의 숫자들을 리스트에 저장하고, 생성자이면 리스트에서 삭제 한 후 리스트를 출력하는 방법으로 문제를 해결하였다.

 

매개변수로 num과 리스트 a를 받는 check함수를 정의하였다.

for n in str(num)에서 num을 문자열로 바꿔준 뒤 한 자리씩 더하였다.

num이 10000보다 작고, num이 a에 존재하는 경우 a에서 num을 지운다.

a.count(num) > 0을 하지 않으면 생성자가 2개인 경우, 리스트에 존재하지 않는 요소를 삭제하게 되어 오류가 발생한다.

def check(num, a: list):
    for n in str(num):
        num += int(n)
    
    if num <= 10000 and a.count(num) > 0:
        a.remove(num)

 

main에서는 1부터 10000을 요소로 가진 리스트 selfNumber를 정의 해주었다.

첫 번째 for문에서는 1부터 10000까지 생성자인지 확인하고, 삭제하는 함수 check을 호출한다.

두 번째 for문에서는 셀프 넘버만 남은 리스트를 출력한다.

if __name__ == '__main__':
    selfNumber = list(range(1,10001))
    
    for i in range(1, 10000):
        check(i, selfNumber)

    for i in selfNumber:
        print(i)