[백준] 함수 - 4673번 (Python)

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

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보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 증가하는 순서로 출력한다.

코드 정답

list 활용 코드

numbers = list(range(1, 10001))
remove_list = []  # 이후에 삭제할 숫자 list
for num in numbers :
    for n in str(num):
        num += int(n)  # 생성자가 있는 숫자
    if num <= 10000:  # 10,000보다 작거나 같을 때만,
        remove_list.append(num)  # append: 리스트에 요소를 추가할 때

for remove_num in set(remove_list) :  # set 으로 중복값 제거
    numbers.remove(remove_num)
for self_num in numbers :  # 생성자가 있는 숫자를 삭제한 리스트
    print(self_num)

set 활용 코드

numbers = set(range(1, 10000))
remove_set = set()  # 생성자가 있는 숫자 set
for num in numbers :
    for n in str(num):
        num += int(n)
    remove_set.add(num)  # add: 집합에 요소를 추가할 때

self_numbers = numbers - remove_set  # set의 '-' 연산자로 차집합을 구함
for self_num in sorted(self_numbers):  # sorted 함수로 정렬
    print(self_num)

풀이

< list 활용 코드 >

1행 : 1부터 10,000 까지의 수를 numbers 변수에 리스트 형식으로 저장한다.

2행 : remove_list 라는 변수를 리스트 형식으로 생성한다.

3행 : num 이 numbers 에 저장된 값을 따라갈 때까지 4~7행을 반복한다.

4행 : n 이 num 에 저장된 값의 길이에 다다를 때까지 5행을 반복한다.

5행 : num = num + n 을 실행한다.

6행 : 만약 num 의 값이 10,000 보다 작거나 같을 경우 7행을 실행한다.

7행 : num 에 저장된 값을 remove_list 에 추가한다.

9행 : remove_num 이 중복값이 제거된 remove_list 에 저장된 값을 따라갈 때까지 10행을 반복한다.

10행 : remove_num 에 있는 값을 numbers 에서 제거한다.

11행 : self_num 이 numbers 에 저장되어 있는 값을 다 따라갈 때까지 12행을 반복한다.

12행 : self_num 을 출력한다.

 

< set 활용 코드 >

1행 : 1부터 10,000 까지의 수를 numbers 변수에 집합 형식으로 저장한다.

2행 : remove_set 라는 변수를 집합 형식으로 생성한다.

3행 : num 의 값이 numbers 에 저장된 값을 따라갈 때까지 4~6행을 반복한다.

4행 : n 의 값이 num 의 길이에 다다를 때까지 5행을 반복한다.

5행 : num = num + int(n) 을 실행한다.

6행 : remove_set 에 num 값을 추가한다.

8행 : numbers 에서 remove_set 의 값의 차집합을 구해 self_numbers 에 저장한다.

9행 : self_num 이 정렬된 self_numbers 의 값에 다다를 때까지 10행을 반복한다.

10행 : self_num 을 출력한다.