[21919] 소수 최소 공배수 (Python)

[문제]

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

21919번: 소수 최소 공배수

---

알고리즘 분류는 수학, 정수론, 소수 판정, 에라토스테네스의 체 입니다.

중복된 소수가 나올 수 있습니다. (이 경우를 고려하지 않으면 출력 초과 오류가 뜹니다.)

---

[코드]

import sys


def prime():
    n = 1000000
    p = [True] * n
    m = int(n ** 0.5)
    for i in range(2, m + 1):
        if p[i] == True:
            for j in range(i + i, n, i):
                p[j] = False
    return p


N = int(sys.stdin.readline())
A = map(int, sys.stdin.readline().split())
P = prime()

res = 1
cnt = 0
dt = dict()
for i in A:
    if P[i] == True and i not in dt:
        res *= i
        dt[i] = 0
        cnt += 1

if cnt != 0:
    print(res)
else:
    print(-1)