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

• 問題: No.3030 ミラー・ラビン素数判定法のテスト
• ユーザー: fura
• 提出日時: 2020-12-11 17:37:47
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 856 bytes
• コンパイル時間: 2,074 ms
• コンパイル使用メモリ: 201,972 KB
• 実行使用メモリ: 6,824 KB
• 最終ジャッジ日時: 2024-11-18 18:59:14
• 合計ジャッジ時間: 2,680 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,820 KB
testcase_01	AC	2 ms; 6,820 KB
testcase_02	AC	2 ms; 6,816 KB
testcase_03	AC	2 ms; 6,820 KB
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-

ソースコード

#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,int m){
	lint r=1%m;
	for(lint x=a%m;k>0;k>>=1,x=x*x%m) if(k&1) r=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;
}