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

• 問題: No.3030 ミラー・ラビン素数判定法のテスト
• ユーザー: fura
• 提出日時: 2020-12-11 18:12:17
• 言語: C++17 (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 420 ms / 9,973 ms
• コード長: 789 bytes
• コンパイル時間: 2,290 ms
• コンパイル使用メモリ: 198,924 KB
• 実行使用メモリ: 4,384 KB
• 最終ジャッジ日時: 2023-08-10 16:27:22
• 合計ジャッジ時間: 3,568 ms
• ジャッジサーバーID （参考情報）: judge11 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,384 KB
testcase_01	AC	1 ms; 4,380 KB
testcase_02	AC	1 ms; 4,384 KB
testcase_03	AC	2 ms; 4,380 KB
testcase_04	AC	219 ms; 4,380 KB
testcase_05	AC	207 ms; 4,384 KB
testcase_06	AC	57 ms; 4,380 KB
testcase_07	AC	56 ms; 4,384 KB
testcase_08	AC	57 ms; 4,384 KB
testcase_09	AC	420 ms; 4,380 KB

ソースコード

#include <bits/stdc++.h>

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

using namespace std;
using lint=long long;

bool Miller_Rabin(lint n){
	if(n<=1)   return false;
	if(n%2==0) return n==2;

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

	int r=0;
	lint d=n-1;
	while(d%2==0) r++, d>>=1;

	for(lint a:{2,3,5,7,11,13,17,19,23,29,31,37}){
		if(a>=n-2) break;
		lint x=modpow(a,d,n);
		if(x==1 || x==n-1) continue;
		bool b=false;
		rep(_,r-1){
			x=__int128(x)*x%n;
			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;
}