다음 코드는 N 이 소수이면 YES, 아니면 NO 를 출력한다. 구현은 옳지만 N 이 커지면 시간 초과가 난다. 같은 결과를 시간 제한 안에서 출력하라.
def is_prime(n):
if n < 2: return False
for i in range(2, n):
if n % i == 0: return False
return True
n = int(input())
print('YES' if is_prime(n) else 'NO')
0 ≤ N ≤ 10^9
2
YES
4
NO
17
YES