파이썬 / BOJ / 1037번 (약수)

문제

양수 A가 N의 진짜 약수가 되려면, N이 A의 배수이고, A가 1과 N이 아니어야 한다. 어떤 수 N의 진짜 약수가 모두 주어질 때, N을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N의 진짜 약수의 개수가 주어진다. 이 개수는 50보다 작거나 같은 자연수이다. 둘째 줄에는 N의 진짜 약수가 주어진다. 1,000,000보다 작거나 같고, 2보다 크거나 같은 자연수이고, 중복되지 않는다.

출력

첫째 줄에 N을 출력한다. N은 항상 32비트 부호있는 정수로 표현할 수 있다

풀이

들어오는 약수들을 오름차순으로 정리한 다음,

약수의 갯수가 짝수의 경우에는 맨 처음과 맨 끝(즉, 가장 작은 값과 가장 큰 값)을 곱하였다.

홀수의 경우에는 중간의 값 (즉, 약수에서 중복되는 경우)의 값을 두번 곱하였다.

import sys


N = int(sys.stdin.readline())
aliquot = list(map(int,sys.stdin.readline().split()))
aliquot = sorted(aliquot)
#짝수 일 때
if N % 2 == 0:
    print(aliquot[0]*aliquot[-1]) 
#홀수 일 때
elif N % 2 == 1:
    print(aliquot[(N-1)//2]**2)

 

하지만 이렇게 굳이 하지 않아도 홀수건 짝수건 가장 큰 값과 가장 작은 값을 곱한 값을 출력해주면 답이 된다.

import sys
sys.stdin.readline()
a = list(map(int, sys.stdin.readline().split()))
print(max(a)*min(a))