[BOJ_Python] 21919_소수 최소 공배수

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

21919번: 소수 최소 공배수

import sys

n = int(input())
lst = list(map(int, sys.stdin.readline().split()))

def od(a):
    if a == 1:
        return 0
    else:
        for i in range(2, a):
            if a % i == 0:
                return 0
        return 1

ans_lst = []
for k in lst:
    if od(k) == 1:
        ans_lst.append(k)

if len(ans_lst) == 0:
    print(-1)
else:
    cnt = 1
    for k in ans_lst:
        cnt *= k
    print(cnt)

틀린 이유 : 시간 초과가 나서 뭐가 문제일까 생각을 했는데 중간에 소수를 구해서 소수의 최소 공배수를 구할 때, 소수를 빈 리스트에 넣고 그 리스트의 원소 전부를 곱하는 방식으로 코드를 짰다. 근데 보니까 굳이 이렇게 돌아갈 필요 없이 수열의 수들을 소수인지 판별하고 소수이면 1에 곱해가는 방식으로 간단하게 풀었어야 했다.

 + 수열의 중복을 없애야하기 때문에 list 말고 set으로 했어야 했다.

 

코드

import sys
import math

n = int(input())
lst = set(map(int, sys.stdin.readline().split()))

def od(a):
    if a == 1:
        return 0
    else:
        for i in range(2, a):
            if a % i == 0:
                return 0
        return 1

ans = 1
for k in lst:
    if od(k) == 1:
        ans = ans * k

if ans == 1:
    print(-1)
else:
    print(ans)