[Python] 백준 24039번(2021은 무엇이 특별할까?) 문제 풀이

 

import math

def is_prime(n):
    if n == 1:
        return False
    for i in range(2, int(math.sqrt(n))+1):
        if n % i == 0:
            return False
    return True
    
n = int(input())

i = 2
ans = []
while True:
    if is_prime(i):
        ans.append(i)
        if len(ans) == 2:
            if ans[0]*ans[1] > n:
                print(ans[0]*ans[1])
                break
            del ans[0]
    if i == 2:
        i += 1
        continue
    i += 2