Python

[Python] 백준 boj 4673 "셀프넘버" - 시간초과 해결

민돌v 2021. 10. 21. 15:10

셀프 넘버 성공출처다국어

한국어   

시간 제한메모리 제한제출정답맞은 사람정답 비율
1 초 256 MB 97988 48847 38728 49.539%

문제

셀프 넘버는 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 복사

예제 출력 1 복사

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

 

 

풀이

처음에는 셀프넘버가 현재의 수와 각 자리수를 더하면 된다는 생각으로

self_number += self_number/10 + self_number %10

이런식으로 직접 재귀함수의 매개변수로 넣어주었다...

 

3자리수 이상은 생각하지 않았던거지... ㅎㅎ

 

그 후에 파이썬의 list 는 문자열을 하나씩 끊어서 요소로 저장한다는 성질을 이용해 각자리수를 분리한 후 정수형으로 더해줫다.

self_number = []


def get_self_number(number):
    if number > 10000:
        return

    number_diget = list(str(number))
    for i in number_diget:
        number = number + int(i)

    if number in self_number:
        self_number.remove(number)
    get_self_number(number)


# main
for i in range(1, 10001):
    self_number.append(int(i))

for num in self_number:
    # 생성자
    print(num)
    get_self_number(num)

 

결과 : 시간초과... 재귀로 풀어서 그런가보다 하구, 효울을 조금 더 올려주었다.

 

self_number = []

# main
for i in range(1, 10001):
    self_number.append(int(i))

for num in self_number:
    # 생성자
    print(num)

    while num < 10000:
        num_digit = list(str(num))
        for i in num_digit:
            num = num + int(i)

        if num in self_number:
            self_number.remove(num)

 

결과 : 이것도 시간초과... 뭔가 이상하다.. 흐음...

더 줄이면 줄일 수 있겠지만, 이이상 줄이는게 의미가 있나 싶었다.

혹시 몰라 pypy3 로 제출하니까 성공...

ㅎㅎㅎㅎㅎ

파이파이로 제출하세요 ㅎㅎㅎ