Các bạn pro c++ giúp mình giải câu này với:) nhưng code ngắn thôi vì nó quy định tg á

`text{CODE}`

#include <bits/stdc++.h>
using namespace std;
long long thu(long long x, long long y, long long p) {
    long long result = 1;
    x = x % p;
    while (y > 0) {
        if (y & 1) {
            result = (result * x) % p;
        }
        y = y >> 1;
        x = (x * x) % p;
    }
    return result;
}
bool kt(long long n, int k) {
    if (n <= 1) return false;
    if (n <= 3) return true;
    if (n % 2 == 0) return false;
    long long m = n - 1;
    int s = 0;
    while (m % 2 == 0) {
        m /= 2;
        s++;
    }
    for (int i = 0; i < k; i++) {
        long long a = 2 + rand() % (n - 3);
        long long x = thu(a, m, n);
        if (x == 1 || x == n - 1) continue;
        for (int j = 0; j < s - 1; j++) {
            x = thu(x, 2, n);
            if (x == n - 1) break;
        }
        if (x != n - 1) return false;
    }

    return true;
}
int main() {
    srand(time(NULL));
    long long n;
    cin >> n;
    if (kt(n, 5))
        cout << "YES";
     else
        cout << "NO";


    return 0;
}

\begin{array}{c} \color{#db1616}{\texttt{#Khoadang09}} \end{array}

// Không sàng vẫn AC mới ảo :))

// Code cũ, gửi lại

#include <bits/stdc++.h>
#define ll long long 
#define nhanh  ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
using namespace std;
bool prime(ll n){
    if (n<2) return false ;
    for (ll i=2;i*i<=n;i++) if (n%i==0) return false ;
    return true ;
}
int main(){
    nhanh ;
    ll n ; cin >> n ;
    (prime(n)) ? (cout<<"YES") : (cout<<"NO");
}