[백준/BOJ] 9506번: 약수들의 합 (Python 파이썬)

문제

: 어떤 숫자 n이 자신을 제외한 모든 약수들의 합과 같으면, 그 수를 완전수라고 한다.

예를 들어 6은 6 = 1 + 2 + 3 으로 완전수이다.

n이 완전수인지 아닌지 판단해주는 프로그램을 작성하라.

 

입출력 규칙

1. 입력

입력은 테스트 케이스마다 한 줄 간격으로 n이 주어진다. (2 < n < 100,000)

입력의 마지막엔 -1이 주어진다.

2. 출력

테스트케이스 마다 한줄에 하나씩 출력해야 한다.

n이 완전수라면, n을 n이 아닌 약수들의 합으로 나타내어 출력한다(예제 출력 참고).

이때, 약수들은 오름차순으로 나열해야 한다.

n이 완전수가 아니라면 n is NOT perfect. 를 출력한다.

 

---

문제 풀이

<내 풀이>

 

• 입력이 -1로 들어올 때까지 계속 실행할 수 있도록 while문을 사용하였다. 
• n의 약수들을 저장할 수 있는 리스트 tmp를 만들었다.
• tmp에 저장된 약수들의 합 (sum(tmp))가 n와 같아 완전수인지 아닌지 확인한다.
• 완전수라면, tmp에 저장되어 있던 약수들을 '+'를 포함해서 출력해야한다.
• 이, 문자열을 합칠 수 있는 join()를 사용하였으며, tmp내에 있는 약수들은 int형이므로, 이를 일괄적으로 str형으로 변환하기 위해 map함수를 사용하였다.

while True:
    n = int(input())

    if n == -1: break

    tmp=[]
    for i in range(1,n):
        if n % i ==0:
            tmp.append(i)
    if sum(tmp) == n :
        print(f"{n} = {' + '.join(map(str,tmp))}")
    else:
        print(f"{n} is NOT perfect.")

 

<숏코딩>

 

숏코딩이다. 진짜 간결하게 구현된 코드이다. 

• [n==sum(tmp)]는 n이 tmp 리스트의 합과 같은지 비교한다.
• n == sum(p)가 True면 인덱스 1의 문자열을 선택하고, False면 인덱스 0의 문자열을 선택한다.

while(n:=int(input()))>0:
    tmp = [i for i in range(1,n) if n%i<1]
    print(n,['is NOT perfect.','= '+' + '.join(map(str,tmp))][n==sum(tmp)])