改进Miller_rabin素数测试

定义：一个合数n，若对所有满足gcd(a,n)=1的正整数a都有a^(n-1)≡1(mod n)成立，则称之为卡迈克尔数。

二次探测定理：如果p是一个素数，且0<x<p，则方程x^2%p=1的解为x=1或x=p-1。

     那么可以根据二次探测定理，在利用费马小定理计算a^(n-1)%n的过程中增加对整数n的二次探测，一旦发现违背二次探测条件，即得出n不是素数的结论。

代码:

#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;
#define TIMES 5
typedef long long LL;

LL random(LL n)
{
	return (LL)((double) rand()/RAND_MAX*n+0.5);
}

LL multi(LL a,LL b,LL mod) //a*b%mod
{
	LL ans=0;
	while(b)
	{
		if(b)
			ans=(ans+a)%mod;
		b>>=1;
		a=(a<<1)%mod;
	}
	return ans;
}

LL kpow(LL a,LL n,LL mod) //a^n%mod
{
	LL ans=1;
	while(n)
	{
		if(n&1)
			ans=(ans*a)%mod;
		n>>=1;
		a=(a*a)%mod; 
	}
	return ans;
}

bool witness(LL a,LL n)
{
	LL m=n-1;
	LL j=0;
	while(! (m&1))
	{
		j++;
		m>>=1;
	}
	LL x=kpow(a,m,n);
	if(x==1 || x==n-1)
		return false;
	while(j--)
	{
		x=x*x%n;
		if(x==n-1)
			return false;
	}
	return true;
}

bool miller_rabin(LL n)
{
	if(n<2) return false;
	if(n==2) return true;
	if(!(n&1)) return false;
	for(LL i=1;i<=TIMES;i++)
	{
		LL a=random(n-2)+1;
		if(witness(a,n))
			return false;
	}
	return true;
}

int main()
{
	LL x;
	while(cin>>x)
	{
		if(miller_rabin(x))
			cout<<"yes\n";
		else
			cout<<"no\n";
	}
	return 0;
}