[파이썬]SelfNumber(generater)들의 합 구하기

어떤 자연수 n이 있을 때, d(n)을 n의 각 자릿수 숫자들과 n 자신을 더한 숫자라고 정의하자.
예를 들어

d(91) = 9 + 1 + 91 = 101

이 때, n을 d(n)의 제네레이터(generator)라고 한다. 위의 예에서 91은 101의 제네레이터이다.
어떤 숫자들은 하나 이상의 제네레이터를 가지고 있는데, 101의 제네레이터는 91 뿐 아니라 100도 있다. 그런데 반대로, 제네레이터가 없는 숫자들도 있으며, 이런 숫자를 인도의 수학자 Kaprekar가 셀프 넘버(self-number)라 이름 붙였다. 예를 들어 1,3,5,7,9,20,31 은 셀프 넘버 들이다.
1 이상이고 5000 보다 작은 모든 셀프 넘버들의 합을 구하라.

listA = list(range(1, 5001))
listB = []

for i in range(1, 5001):
    if i<10:
        j = i + i 
    elif i<100:
        j = int(str(i)[0]) + int(str(i)[1]) + i 
    elif i<1000:
        j = int(str(i)[0]) + int(str(i)[1]) + int(str(i)[2]) + i 
    else:
        j = int(str(i)[0]) + int(str(i)[1]) + int(str(i)[2]) + int(str(i)[3]) + i

    listB.append(j)

setA = set(listA)
setB = set(listB)
listC = list(setA - setB)

sum= 0
for a in listC:
    sum+=a
print(sum)

넥슨 입사문제 중에서 - 코딩도장 (codingdojang.com)