[코테/백준] Python 셀프 넘버 - 4673번

성공 | 시도 2번

 

🔗 [백준] 4673번 : 셀프 넘버

4673 : [브루트포스 알고리즘] 셀프 넘버(py) 

---

문제 설명

셀프 넘버는 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보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 출력하는 프로그램을 작성하시오.

---

문제 분석

• d(n) = n + 각 자리수 

- 먼저 1 ~ 10,000 범위에 해당하는 d(n)의 생성자를 구해 리스트에 담는다. 그 후, 1 ~ 10,000 범위 중에 d(n)의 생성자 리스트에 포함되지 않은 숫자의 개수를 센다. 

---

입력

입력은 없다.

---

출력

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

---

입력과 출력 예시

입력	출력
없음	1 3 5 7 9 20 31 42 53 64  |  |       <-- a lot more numbers  | 9903 9914 9925 9927 9938 9949 9960 9971 9982 9993

 

---

내 소스

def d(n):
    result = n + sum(list(map(int, str(n))))
    return result
array = []
for i in range(1, 10001):
    array.append(d(i))
    
for i in range(1, 10001):
    if i in array:
        continue
    print(i)

 

소스 설명

def d(n):
    result = n + sum(list(map(int, str(n))))
    return result

[1 - 3] 

- sum(list(map(int, str(n)))): n의 각 자리수를 더한다.

• str은 iterable이기 때문에 정수 n을 str으로 변환하고 각 문자를 int으로 변환하여 list에 담는다. 그 후, 리스트 원소들의 합을 구한다.

- n과 n의 각 자리수를 더한 뒤 result 변수에 담아 반환한다.

 

array = []
for i in range(1, 10001):
    array.append(d(i))

[4 - 6] : 1 ~ 10,000 범위에서 d(n)의 생성자에 해당하는 숫자를 array 리스트에 담는다.

 

for i in range(1, 10001):
    if i in array:
        continue
    print(i)

[8 - 11] : 1 ~ 10,000 범위 중에 array에 있지 않는 숫자만 출력한다.

---

다른 코드 분석

- set() 함수를 사용하는 방법이 있었다. 생성자 set을 만든 후, 1 ~ 10,000 범위 set에서 빼고 오름차순 정렬을 하여 출력한다. 생성자가 2개인 경우가 있어서 set을 이용한 것 같다.

- [4] 님의 풀이를 통해 list 변환없이 sum() 함수를 사용해도 된다는 것을 알았다.

• n = n + sum(map(int, str(n)))
• sum(iterable, /, start=0)

- [6] 님은 각 자리수를 구하기 위해 10으로 나눈 나머지와 몫을 사용하였다. 

---

회고

- map()을 사용하다가 헤갈려서 map()에 대한 글을 정리했다. 🔗 map 정리 

---

수정한 최종 소스

total_set = set(range(1, 10001))
def d(n):
    result = n + sum(map(int, str(n)))
    return result

array = []
for i in range(1, 10001):
    array.append(d(i))
    
result_array = total_set - set(array)
    
for i in sorted(result_array):
    print(i)

- 이와 같이 소스를 수정하니 원래 소스보다 시간이 현저히 줄어든 것을 확인할 수 있었다. 

 

제출 결과

---

Ref. 

1. 위키독스(파이썬) / https://wikidocs.net/16068

2. 문제 풀이 / https://wook-2124.tistory.com/252

3. 문제 풀이 / https://ooyoung.tistory.com/64

4. 문제 풀이 / https://kbwplace.tistory.com/69

5. 공식 문서(파이썬) / https://docs.python.org/ko/3/library/functions.html?highlight=sum#sum 

6. 문제 풀이 / https://parkgihyeon.github.io/boj/boj4673/

 

---

* 잘못된 부분에 대해 댓글 남겨주시면 감사하겠습니다! 😀