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

---

A = []
result = []


for i in range(1, 10001):
    A.append(i)

for i in range(len(A)):
    length = len(str(A[i]))
    data = 0
    for j in range(length):
        data += int(str(A[i])[j])
        if j == length - 1:
            result.append(data + A[i])

real_result = [i for i in A if i not in result]
for i in real_result:
    print(i)

 

코드 설명 :

A = []
result = []

# 1부터 10000까지 수 n을 A에 대입
for i in range(1, 10001):
    A.append(i)

# 1부터 10000까지 반복
for i in range(len(A)):
    # 리스트 A에 있는 각각의 값의 길이 length 선언  
    # 만약 7일 때 1, 82일 때 2, 2022일 때 4 ...
    length = len(str(A[i]))
    # n의 각 자리 수를 더할 변수 data 선언
    data = 0

    # 각각 값의 길이만큼 반복
    for j in range(length):
        # 각 자리 수를 더한 값을 data에 대입
        # 만약 39일 때, 39 + 3 + 9를 하는데 3 + 9한 값을 의미함
        data += int(str(A[i])[j])
        # 맨 마지막 길이일 때, 즉 각 자리 수를 다 더했을 때 조건문 실행
        if j == length - 1:
            # 현재 수 n과 더한 값을 result에 추가
            result.append(data + A[i])

# A에서 result가 없는 값(셀프 넘버)을 추출
real_result = [i for i in A if i not in result]
for i in real_result:
    print(i)

 

결과 :

생각보다 시간이 꽤 많이 걸렸다..
맞춘 다른 사람들의 시간을 보니 100ms 아래도 있어서 코드를 봤더니 엄청 깔끔하고 단순하게 만들었다..
더 열심히 알고리즘 공부해야겠다.