[파이썬, Python] 백준 1037: 약수

문제

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

1037번: 약수

코드

import sys
input = sys.stdin.readline

n = int(input())
num = list(map(int, input().split()))

if n == 1:
    print(num[0]**2)
else:
    print(min(num)*max(num))

설명

일반적인 입력의 경우 num 값의 최솟값과 최댓값의 곱이 정답이다.

예제 2와 같은 경우가 예외적인 케이스였는데, 약수가 1개인 경우 제곱을 해주면 된다.