#209861 (C++14) No.577 Prime Powerful Numbers

• 問題: No.577 Prime Powerful Numbers
• ユーザー: ciel
• 提出日時: 2017-10-14 19:12:13
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,170 bytes
• コンパイル時間: 428 ms
• コンパイル使用メモリ: 40,320 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-17 18:45:09
• 合計ジャッジ時間: 1,394 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	3 ms; 5,248 KB
testcase_01	WA	-
testcase_02	AC	3 ms; 5,248 KB
testcase_03	WA	-
testcase_04	AC	8 ms; 5,248 KB
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	AC	2 ms; 5,248 KB

ソースコード

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
typedef unsigned long long ull;

#if __SIZEOF_POINTER__==8
typedef ull val_t;
val_t po(val_t x,val_t y,val_t m){
	val_t z=1;
	for(;y;y>>=1){
		if(y&1)z=z*x%m;
		x=x*x%m;
	}
	return z;
}
#elif __SIZEOF_POINTER__==4
typedef ull val_t;
val_t po(val_t x,val_t y,val_t m){
	val_t z=1;
	for(;y;y>>=1){
		if(y&1)z=z*x%m;
		x=x*x%m;
	}
	return z;
}
#else
#error modpow is not supported
#endif

ull gcd(ull x,ull y){return y?gcd(y,x%y):x;}
bool miller_rabin(ull n){
	if(n==2)return true;
	if(n==1||n%2==0)return false;
	ull d=n-1,s=0,a=1;
	for(;d%2==0;d/=2)s+=1;
	for(int k=5;k--;){ //todo
		for(a++;gcd(a,n)!=1;a++); //todo
		ull r=po(a,d,n);
		if(r==1||r==n-1)continue;
		int t=s;
		for(;t;t--){
			r=po(r,2,n);
			if(r==n-1)break;
		}
		if(!t)return false;
	}
	return true;
}

void solve(ull n){
	if(n%2==0){puts(n>2?"Yes":"No");return;}
	for(ull m=2;m<n;m*=2){
		ull z=n-m;
		for(int k=1;k<61;k++){
			ull r=k==1?z:expl(logl(n+1)/k)+1e-6;
			if(m+po(r,k,1LL<<62)!=n)continue;
			if(miller_rabin(r)){puts("Yes");return;}
		}
	}
	puts("No");
}
int main(){
	ull t,n;
	for(scanf("%llu",&t);t--;solve(n))scanf("%llu",&n);
}