关闭

判断是否为素数

79人阅读 评论(0) 收藏 举报
分类:
#include"iostream"
using namespace std;
long long pow_mod(long long a,long long i,long long n)
{
	if(i==0) return 1;
	int temp=pow_mod(a,i>>1,n);
	temp=temp*temp%n;
	if(i&1) temp=( long long ) temp*a%n;
	return temp;
}
bool test(int n,int a,int d)
{
	if(n==2) return true;
	if(n==a) return true;
	if((n&1)==0) return false; //n是偶数
	while(!(d&1)) d=d>>1;	  //当d为奇数进行
	int t=pow_mod(a,d,n);
	while((d!=n-1)&&(t!=1)&&(t!=n-1))
	{
		t=(long long) t*t%n;
		d=d<<1;
	}
	return (t==n-1||(d&1)==1);
}
bool isprime(int n)
{
	if(n<3) return false ;
	int a[]={2,3,61};
	for(int i=0;i<=2;i++)
		if(!test(n,a[i],n-1))return false;
	return true;
}
int main()
{
	int n;
	while(cin>>n)
	{
		if(n==2) cout<<"YES\n";
		else
		{
		if(isprime(n))
			cout<<"YES\n";
		else 
			cout<<"NO\n";
		}
	}
	return 0;
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:34307次
    • 积分:2750
    • 等级:
    • 排名:第13534名
    • 原创:249篇
    • 转载:20篇
    • 译文:0篇
    • 评论:0条