#592173 (C++17) No.3030 ミラー・ラビン素数判定法のテスト

• 問題: No.3030 ミラー・ラビン素数判定法のテスト
• ユーザー: fura
• 提出日時: 2020-12-11 17:42:34
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 685 ms / 9,973 ms
• コード長: 879 bytes
• コンパイル時間: 2,176 ms
• コンパイル使用メモリ: 202,092 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-16 23:35:03
• 合計ジャッジ時間: 4,426 ms
• ジャッジサーバーID （参考情報）: judge2 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	2 ms; 5,248 KB
testcase_04	AC	344 ms; 5,248 KB
testcase_05	AC	318 ms; 5,248 KB
testcase_06	AC	51 ms; 5,248 KB
testcase_07	AC	50 ms; 5,248 KB
testcase_08	AC	48 ms; 5,248 KB
testcase_09	AC	685 ms; 5,248 KB

ソースコード

#include <bits/stdc++.h>

#define rep(i,n) for(int i=0;i<(n);i++)

using namespace std;
using lint=long long;

random_device seed_gen;
mt19937 rng(seed_gen());

lint modpow(lint a,lint k,lint m){
	lint r=1%m;
	for(lint x=a%m;k>0;k>>=1,x=__int128(x)*x%m) if(k&1) r=__int128(r)*x%m;
	return r;
}

bool Miller_Rabin(lint n,int k=20){
	if(n<=1)   return false;
	if(~n&1)   return n==2;
	if(n%3==0) return n==3;

	int s=0; // n-1 = 2^s * d (d: odd)
	lint d=n-1;
	while(~d&1) s++, d>>=1;

	while(k--){
		lint a=rng()%(n-3)+2;
		lint x=modpow(a,d,n);
		if(x==1 || x==n-1) continue;
		bool b=false;
		for(int r=1;r<s;r++){
			x=__int128(x)*x%n;
			if(x== 1 ) return false;
			if(x==n-1){ b=true; break; }
		}
		if(!b) return false;
	}

	return true;
}

int main(){
	int q; scanf("%d",&q);
	rep(_,q){
		lint x; scanf("%lld",&x);
		printf("%lld %d\n",x,!!Miller_Rabin(x));
	}
	return 0;
}