x^n + y^n = z^n (mod p)
假设p的原根是r
我们设 x=r^a,y=r^b,z=r^c;
原方程化为
r^an + r^bn = r^cn (mod p)
令R=r^n
设a<=b
=> 1 + R^(b-a) = R^(c-a) (mod p) (c>=a)
或者
R^(a-c) + R^(b-c) = 1 (mod p) (c<a)
我们枚举R^k
若有1+R^k1=R^k2 或 R^k1+R^k2=1则有解
枚举时若R^k出现循环则break。
CODE:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define LL int
#define nMax 1000001
int a[nMax],l;
inline int pow_mod(int a,int n,int mod){
long long y=1LL,b=(long long)a;
while (n>0){
if (n&1) y=y*b%mod;
n>>=1;
b=b*b%mod;
}
return (int)y;
}
inline int g_test(int g,int p){
for(register LL i=0;i<l;i++) if(pow_mod(g,(p-1)/a[i],p)==1) return 0;
return 1;
}
int primitive_root(int p){
int tmp=p-1;
l=0;
for(register int i=2;i<=tmp/i;i++) if(tmp%i==0){
a[l++]=i;
a[l++]=tmp/i;
//while(tmp%i==0) tmp/=i;
}
int g=1,j;
while(1){
for(j=0;j<l;j++) if(pow_mod(g,tmp/a[j],p)==1) break;
if(j>=l) return g;
g++;
}
}
int A[nMax]={0},B[nMax];
int n,p;
int x,y,z,s,R,ok,s1,s2;
int j;
int r;
#define bug puts("Fuck");
int main(){
#ifndef ONLINE_JUDGE
freopen("input.txt","r",stdin);
// freopen("output.txt","w",stdout);
#endif
int t;
int cas=1;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&p);
r=primitive_root(p);
R=pow_mod(r,n,p);
ok=0;
s=1;
//A[1]=cas;B[1]=0;
for(register LL i=0;i<p && !ok;i++){
if(i) s=(long long)s*R%p;
if(A[s]==cas) break;
A[s]=cas;B[s]=i;
s1=1+s;if(s1>=p) s1=0;
s2=1-s+p;while(s2>=p) s2=0;
if(A[s1]==cas){
j=B[s1];
x=1,y=pow_mod(r,i,p),z=pow_mod(r,j,p);
ok=1;break;
}else if(A[s2]==cas){
j=B[s2];
x=pow_mod(r,i,p),y=pow_mod(r,j,p),z=1;
ok=1;break;
}
//A[s]=cas;B[s]=i;
}
if(ok) printf("%d %d %d\n",x,y,z);
else printf("-1\n");
cas++;
}
return 0;
}