算法提高 素数判断

编写一函数IsPrime，判断某个大于2的正整数是否为素数。

样例输入: 
5
样例输出:
yes

样例输入: 
9
样例输出:
no

注意：是素数输出yes，不是素数输出no，其中yes和no均为小写。

#include<iostream>
#include<cmath>
#include <iostream>
#include <iomanip>
using namespace std;


int isPrime(long long n)
{	//返回1表示判断为质数，0为非质数，在此没有进行输入异常检测
    float n_sqrt;
    if(n==2 || n==3) return 1;
    if(n%6!=1 && n%6!=5) return 0;
    n_sqrt = floor(sqrt((float)n));
    for(int i=5; i<=n_sqrt; i+=6)
    {
        if(n%(i)==0 | n%(i+2)==0) return 0;
    }
    return 1;
}



int main()
{
    int n;
    cin >> n;

    if(isPrime(n))
        cout << "yes" << endl;
    else cout << "no" << endl;
    
    return 0;
}

 

//判断某一区间内的素数个数
void segment_sieve(ll a, ll b) {
	for (int i = 0; (ll)i * i < b; i++) is_prime_small[i] = true;
	for (int i = 0; i < b - a; i++) is_prime[i] = true;
	
	for (int i = 2; (ll)i * i < b; i++) {
		if (is_prime_small[i]) {
			for (int j = 2*i; (ll)j*j < b; j += i) is_prime_small[j] = false;
			for (ll j = max(2LL, (a+i-1)/i)*i; j < b; j += i) {
				is_prime[j-a] = false;
			}
		}
	}
	return;
}