結果

問題 No.981 一般冪乗根
ユーザー 37zigen37zigen
提出日時 2020-03-14 19:16:32
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 11 ms / 6,000 ms
コード長 2,044 bytes
コンパイル時間 1,169 ms
コンパイル使用メモリ 96,152 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-05-03 01:27:52
合計ジャッジ時間 43,851 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
5,248 KB
testcase_01 AC 4 ms
5,376 KB
testcase_02 AC 5 ms
5,376 KB
testcase_03 AC 4 ms
5,376 KB
testcase_04 AC 4 ms
5,376 KB
testcase_05 AC 4 ms
5,376 KB
testcase_06 AC 4 ms
5,376 KB
testcase_07 AC 4 ms
5,376 KB
testcase_08 AC 4 ms
5,376 KB
testcase_09 AC 4 ms
5,376 KB
testcase_10 AC 4 ms
5,376 KB
testcase_11 AC 4 ms
5,376 KB
testcase_12 AC 4 ms
5,376 KB
testcase_13 AC 4 ms
5,376 KB
testcase_14 AC 4 ms
5,376 KB
testcase_15 AC 4 ms
5,376 KB
testcase_16 AC 4 ms
5,376 KB
testcase_17 AC 4 ms
5,376 KB
testcase_18 AC 4 ms
5,376 KB
testcase_19 AC 4 ms
5,376 KB
testcase_20 AC 4 ms
5,376 KB
testcase_21 AC 4 ms
5,376 KB
testcase_22 AC 4 ms
5,376 KB
testcase_23 AC 4 ms
5,376 KB
testcase_24 AC 4 ms
5,376 KB
testcase_25 AC 6 ms
5,376 KB
testcase_26 AC 5 ms
5,376 KB
testcase_27 AC 3 ms
5,376 KB
testcase_28 AC 11 ms
5,376 KB
evil_60bit1.txt WA -
evil_60bit2.txt WA -
evil_60bit3.txt WA -
evil_hack AC 2 ms
5,376 KB
evil_hard_random WA -
evil_hard_safeprime.txt WA -
evil_hard_tonelli0 WA -
evil_hard_tonelli1 WA -
evil_hard_tonelli2 WA -
evil_hard_tonelli3 WA -
evil_sefeprime1.txt WA -
evil_sefeprime2.txt WA -
evil_sefeprime3.txt WA -
evil_tonelli1.txt WA -
evil_tonelli2.txt WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cassert>
#include <cmath>
#include <map>

long long pow(long long a, long long n, long long p) {
	long long r=1;
	for (;n>0;n>>=1,a=a*a%p)if(n%2==1)r=r*a%p;
	return r;
}

int cnt(long long a,long long base,long long p) {
	int ret=0;
	while (a!=1) {
		a=pow(a,base,p);
		++ret;
	}
	return ret;
}

long long inv(long long a, long long p) {
	a%=p;
	long long u=1, v=0;
	long long b=p;
	while (b>0) {
		long long q=a/b;
		a%=b;
		u-=v*q%p;
		u=(u%p+p)%p;
		{
			u^=v;v^=u;u^=v;
			a^=b;b^=a;a^=b;
		}
	}
	return u<0?u+p:u;
}

long long gcd(long long a,long long b) {
	return a==0?b:gcd(b%a,a);
}
	
long long peth_root(long long a,long long p,int e,long long mod) {
	long long q=mod-1;
	int s=0;
	while (q%p==0) {
		q/=p;
		++s;
	}
	long long pe=pow(p,e,mod);
	long long ans=pow(a,((pe-1)*inv(q,pe)%pe*q+1)/pe,mod);
	long long c=2;
	while (pow(c,(mod-1)/p,mod)==1)
		++c;
	c=pow(c,q,mod);
	std::map<long long,int> map;
	long long add=1;
	int v=(int)std::sqrt((double)(s-e)*p)+1;
	long long mul=pow(c,v*pow(p,s-1,mod-1)%(mod-1),mod);
	for (int i=0;i<=v;++i) {
		map[add]=i;
		add=add*mul%mod;
	}
	mul=inv(pow(c,pow(p,s-1,mod-1),mod),mod);
	for (int i=e;i<s;++i) {
		long long err=inv(pow(ans,pe,mod),mod)*a%mod;
		long long target=pow(err,pow(p,s-1-i,mod-1),mod);
		for (int j=0; j<=v; ++j) {
			if (map.find(target)!=map.end()) {
				int x=map[target];
				ans=ans*pow(c,(j+v*x)*pow(p,i-e,mod-1)%(mod-1),mod)%mod;
				break;
			}
			target=target*mul%mod;
			assert(j!=v);
		}
	}
	return ans;
}
	
long long kth_root(long long a,long long k,long long p) {
	long long g=gcd(k,p-1);
	if (pow(a,(p-1)/g,p)!=1)
		return -1;
	a=pow(a,inv(k/g,(p-1)/g),p);
	for (long long div=2;div*div<=g;++div) {
		int sz=0;
		while (g%div==0) {
			g/=div;
			++sz;
		}
		if (sz>0) {
			long long b=peth_root(a,div,sz,p);
			a=b;
		}
	}
	if (g>1)
		a=peth_root(a,g,1,p);
	return a;
}

int main() {
	int T;
	scanf("%d",&T);
	for(int t=0;t<T;++t){
		long long p,k,a;
		scanf("%lld%lld%lld",&p,&k,&a);
		printf("%lld\n",kth_root(a, k, p));
	}
}
0