基于单点判定的方法来判定素数

素数指的就是，该数不可以被除了该数本身和1其他的数整除。
本方法对每一个数都进行了判定。
先输入数字N，然后再数入N个数字，每次输入一个数字，都判定该数字是不是素数。如果是素数，就输出YES，否则输出NO。

#include <iostream>
#include <cmath>
using namespace std;
bool is_prime(int n){
    if(n <= 1) return false;
    int sqr = (int)sqrt(n*1.0);
    for(int i = 2;i <= sqr;i ++)
        if(n % i == 0) return false;
    return true;
}
int main(){
    int n,one;
    cin >> n;
    for(int i = 0;i < n;i ++){
        cin >> one;
        (is_prime(one))? cout << "YES" << endl : cout << "NO" << endl ;
    }
    return 0;
}
/*
sample input:
5
5 6 7 8 9
sample output:
YES
NO
YES
NO
NO
*/