help me trl đúng hứa vote 5 sao

$\texttt{C++}$

#include <bits/stdc++.h>

using namespace std;

bool check(long long n)
{
    if (n < 2)
        return false;
    for (long long i = 2; i * i <= n; i++)
    {
        if (n % i == 0)
            return false;
    }
    return true;
}

int main()
{
    long long n;
    cin >> n;
    if (check(n))
    {
        cout << "YES";
    }
    else
    {
        cout << "NO";
    }
    return 0;
}

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