9506: 약수들의 합 - Python

https://www.acmicpc.net/problem/9506

문제

어떤 숫자 n이 자신을 제외한 모든 약수들의 합과 같으면, 그 수를 완전수라고 한다.
예를 들어 6은 6 = 1 + 2 + 3 으로 완전수이다.
n이 완전수인지 아닌지 판단해주는 프로그램을 작성하라.

입력

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

출력

테스트케이스 마다 한줄에 하나씩 출력해야 한다.
n이 완전수라면, n을 n이 아닌 약수들의 합으로 나타내어 출력한다(예제 출력 참고).
이때, 약수들은 오름차순으로 나열해야 한다.
n이 완전수가 아니라면 n is NOT perfect. 를 출력한다.

예제 입출력

풀이 1 (정답)

while True:
	n = int(input())
	if n == -1: break
	factors = [i for i in range(1, n) if n % i == 0]

	if n == sum(factors): # n이 완전수라면
		print(n, '= ', end='') # 이하 덧셈 출력
		for f in factors[0:-1]:
			print(f, end=' + ')
		print(factors[-1])
	else: 
    	print(n, 'is NOT perfect.')

완전수인지 아닌지를 파악하기 위해 sum() 함수를 사용하였다.

if n == sum(factors)

어떻게 덧셈을 출력하여야 할 지가 고민이었고, 조금 난잡하게 해결하였다.
입력이 6, 출력이 '6 = 1 + 2 + 3'인 경우를 예로 든다면

print(n, '= ', end='') # 1. '6 = ' 출력
for f in factors[0:-1]: # 2. '1 + 2 + ' 출력
	print(f, end=' + ')
print(factors[-1]) # 3. '3' 출력

위의 과정으로 덧셈의 출력을 구현하였다.

풀이 2 (정답, unpacking)

while True:
	n = int(input())
	if n == -1: break
	factors = [i for i in range(1, n) if n % i == 0]

	if n == sum(factors):
		print(n, '= ', end='')
		print(*factors, sep = ' + ')
	else: 
		print(n, 'is NOT perfect.')

풀이 1의 덧셈 출력 과정의 for문을 unpacking(*)으로 대체했다.
이 글을 참고하였다.

풀이 3 (정답, join())

while True:
	n = int(input())
	if n == -1: break
	factors = [i for i in range(1, n) if n % i == 0]

	if n == sum(factors):
		print(n, ' = ', ' + '.join(str(i) for i in factors), sep='')
	else: 
		print(n, 'is NOT perfect.')

풀이 1의 덧셈 출력 과정의 for문을 join() 함수로 대체했다.
이 글을 참고하였다.

오늘의 교훈

• list 안의 내용을 전부 꺼내서 출력할 때는 unpacking(*)을 적절히 사용해보자.
• 또는 join() 함수를 사용하는 것도 좋은 방법이 될 수 있다.