파이썬 알고리즘 089 | [백준 4673] 셀프넘버

89. 셀프넘버

셀프 넘버는 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보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 증가하는 순서로 출력한다.
1
3
5
7
9
20
31
42
53
64
|
| <-- a lot more numbers
|
9903
9914
9925
9927
9938
9949
9960
9971
9982
9993

<내 풀이>

(1)====>recursion error

import sys
sys.setrecursionlimit(10**6)
#sys.stdin=open('input.txt','r')

def NotSelfNumber(n):
    nn=list(map(int,str(n)))
    if n>10000:
        return
    elif n>=1000:
        nnn=n+nn[0]+nn[1]+nn[2]+nn[3]
    if n>=100:       
        nnn=n+nn[0]+nn[1]+nn[2]
    elif n>=10:
        nnn=n+nn[0]+nn[1]
    elif n<10:
        nnn=n+n
    if nnn in k:
        k.remove(nnn)
    NotSelfNumber(n+1)  

if __name__=="__main__":
    k=list(range(1,10000))
    NotSelfNumber(1)
    for i in k:
        print(i)

(2)
for문으로 도전

def sf(n):
    k=list(range(1,n+1))
    for i in range(n+1):
        ii=list(map(int,str(i)))
        for j in range(len(ii)):
            i+=ii[j]
        if i in k :
            k.remove(i)
    for i in k:
        print(i)
            
sf(10000)

<풀이>

self_num = [False]+[True]*10000
for number in range(1,10001):
    tmp = number + (number//10000) + (number//1000)%10 + (number//100)%10+ (number//10)%10 + (number) % 10
    if tmp<=10000:  # 생성자가 하나라도 있으면 False
        self_num[tmp] = False

for number, isselfnum in enumerate(self_num):
  if isselfnum:
      print(number)

(2)

def d(n):
    answer = n
    for i in range(len(str(n))):
        answer += int(str(n)[i])
    return answer

arr = set()

i = 1
while d(i) <= 15000:
    arr.add(d(i))
    i += 1


for i in range(1, 10001):
    if i not in arr:
        print(i)

<반성점>

• 논리는 잘 맞는데 recursion limit의 늪에서 빠져나올 방법을 찾지 못하였다 (100까지 하면 옳게 잘 출력됨)
• 왜 나는 setrecursionerror 가 적용되지 않는 걸까요?

<배운 점>

인터프리터에서 직접 실행된 경우에만, if문 이하의 코드를 돌리라는 명령

if __name__ == "__main__":
    //코드
    //코드

모듈을 실행할 수 있는 방법은 직접 실행하거나 임포트하거나
출처 : 링크텍스트

1. input() 말고 sys.stdin.readline() 를 사용하자.

import sys  
a = list(map(int, sys.stdin.readline().split()))

#a = [1, 2, 3, 4, 5]
여러 라인 입력받아야 할 경우, 다음 방법이 빠르다고 한다.

import sys  
n = input()  
a = [sys.stdin.readline() for i in range(n)] 

#a = ["1 2 3", "4 5 6"]

2. 재귀함수가 있는 경우, 최대 재귀 깊이를 설정해줘야 한다.

• • 함수 recursion error : 재귀함수가 1000번 까지만 돌려지기 때문에
    => Maximum recursion depth exceed.. => sys.setrecursionerror(10000)
    이전에 문제풀다가 말한 적이 있는데, 특히 자주접하는 DFS, BFS 문제의 경우 그렇다.
    파이썬의 재귀 허용 깊이의 기본치가 c 보다 약한건지, c 로 짠 DFS, BFS의 경우 문제 없이 돌아가지만, 파이썬으로 돌리면 런타임 오류가 뜨는 경우가 있다. 그래서 다음과 같이 재귀 허용 깊이를 수동으로 늘려주는 코드를, 코드 상단에 적어줘야 한다.

import sys  
sys.setrecursionlimit(10**8) # 10^8 까지 늘림.