結果

問題 No.8030 ミラー・ラビン素数判定法のテスト
ユーザー ok
提出日時 2019-09-04 10:01:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 2,215 bytes
コンパイル時間 755 ms
コンパイル使用メモリ 98,044 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-18 17:02:03
合計ジャッジ時間 3,111 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<string>
#include<iomanip>
#include<cmath>
#include<vector>
#include<algorithm>

#include<ctime>
#include <random>
#include <limits>

using namespace std;

// #define int __int128
#define int long long
#define endl "\n"

// powerのオーバーフローに注意(素数が1~2^63のときには__int128などを使用すること)
//https://primes.utm.edu/prove/prove2_3.html

template<typename Type>
inline Type  product(Type  x, Type  n, Type mod){
	Type  ans = 0;
	while(n){
		if(n&1) ans += x;
		n >>= 1;
		x += x;
		x %= mod;
		ans %= mod;
	}
	return ans;
}

template<typename Type>
inline Type  power(Type  x, Type  n, Type mod){
	Type  ans = 1;
	while(n){
		if(n&1) ans *= x;
		// if(n&1) ans = product(ans,x,mod);
		n >>= 1;
		x *= x;
		// x = product(x,x,mod);
		x %= mod;
		ans %= mod;
	}
	return ans;
}

template<typename Type = __int128>
inline bool Miller_Rabin(Type num, long long k){
	if(num <= 1)        return false;
	else if(num == 2)   return true;
	else if(num%2 == 0) return false; 
	
	Type  s = 0, d = 0;
	vector<int> div;
	
	// if(false){ // num < 2^32
	if(num < (1<<31)){ // num < 2^32
		div = {2,7,61};
	// } else if(num < (1<<63)){
	} else if(true){ // num < 2^63
		div = {2, 325, 9375, 28178, 450775, 9780504, 1795265022};
	} else {
		random_device rnd;
		mt19937 mt(rnd()); 
		// uniform_int_distribution<> rd(1,min((int)numeric_limits<int>::max(), num-1));
		for(long long i = 0; i < k; i++){
			// div.push_back(rd(mt));
			div.push_back(mt()%(num-1)+1);
		}
	}
	
	for(Type n = num-1;;){
		if(n%2){ 
			d = n; break;
		} else s++, n >>= 1;
	}
	
	for(Type  i = 0; i < div.size(); i++){
		Type  a = div[i], n = d, a2 = 1;
		bool flag = true;
		
		if(a >= num) break;
		
		a2 = power(a, d, num);
		
		if(a2 == 1) continue;
		
		for(Type r = 0, two = 1; r < s; r++, two <<= 1){
			a2 = power(a, two*d, num);
			
			if(a2 == num-1){
				flag = false;
				break;
			}
		}
		
		if(flag) return false;
	}
	
	return true;
}

signed main(){
	cin.tie(0);
	ios::sync_with_stdio(false);
	cout<<fixed<<setprecision(10);
	
	int  n;
	
	cin>>n;
	
	for(int i = 0; i < n; i++){
		long long x;
		cin>>x;
		
		cout<<x<<" "<<Miller_Rabin((__int128)x, 10)<<endl;
	}
	
	
	return 0;
}
0