giúp mk với ạ!!!!!!!

`*` Sử dụng thuật toán duyệt căn để kiểm tra tính nguyên tố của 1 số (Bạn có thể tham khảo thuật toán ở 1 số trang web)

$\texttt{Python}$

def isprime(n):
    if n < 2:
        return False
    for i in range(2,int(n**0.5)+1):
        if n%i == 0:
            return False
    return True
n = int(input())
if isprime(n):
    print('YES')
else:
    print('NO')

import math
def p(n):
    if n<2:
        return False
    else:
        a=int(math.sqrt(n))
        for i in range(2,a+1):
            if n % i == 0:
                return False
        return True
n=int(input())
if p(n):
    print("YES")
else:
    print("NO")