소수 판정 가속

다음 코드는 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')