結果

問題 No.981 一般冪乗根
ユーザー tko919tko919
提出日時 2020-04-08 18:28:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,138 bytes
コンパイル時間 1,674 ms
コンパイル使用メモリ 179,172 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-19 02:40:21
合計ジャッジ時間 37,431 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3 WA * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
using namespace std;
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

//template
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define rrep(i,a,b) for(int i=(a);i>(b);i--)
#define ALL(v) (v).begin(),(v).end()
typedef long long int ll;
const int inf = 0x3fffffff; const ll INF = 0x1fffffffffffffff; const double eps=1e-12;
void tostr(ll x,string& res){while(x)res+=('0'+(x%10)),x/=10; reverse(ALL(res)); return;}
template<class T> inline bool chmax(T& a,T b){ if(a<b){a=b;return 1;}return 0; }
template<class T> inline bool chmin(T& a,T b){ if(a>b){a=b;return 1;}return 0; }
//end

ll mpow(ll x,ll k,ll m){
   ll res=1;
   while(k){
      if(k&1)res=(res*x)%m;
      x=(x*x)%m; k>>=1;
   } return res;
}

ll inv(ll a,ll m){
   ll b=m,u=1,v=0;
   while(b){
      ll t=a/b;
      a-=t*b; swap(a,b);
      u-=t*v; swap(u,v);
   }
   u%=m; if(u<0)u+=m; return u;
}

ll sub_root(ll p,int e,ll a,ll m){//x^(p^e)==a(mod m)
   ll q=m-1; int s=0; while(q%p==0){q/=p; s++;} int d=s-e;
   ll pe=mpow(p,e,m),res=mpow(a,((pe-1)*inv(q,pe)%p*q+1)/pe,m),c=1;
   while(mpow(c,(m-1)/p,m)==1)c++; c=mpow(c,q,m);
   map<ll,ll> mp; ll v=1,block=sqrt(d*p)+1,bs=mpow(c,mpow(p,s-1,m-1)*block%(m-1),m);
   rep(i,0,block+1)mp[v]=i,v=v*bs%m;
   ll gs=inv(mpow(c,mpow(p,s-1,m-1),m),m);
   rep(i,0,d){
      ll err=a*inv(mpow(res,pe,m),m)%m;
      ll pos=mpow(err,mpow(p,d-1-i,m-1),m);
      rep(j,0,block+1){
         if(mp.count(pos)){
            res=res*mpow(c,(j+block*mp[pos])*mpow(p,i,m-1)%(m-1),m)%m;
            break;
         } pos=pos*gs%m;
      } 
   } return res;
}

ll mod_root(ll k,ll a,ll m){
   if(a==0)return k?0:-1;
   if(m==2)return a&1;
   k%=m-1; ll g=__gcd(k,m-1);
   if(mpow(a,(m-1)/g,m)!=1)return -1;
   a=mpow(a,inv(k/g,(m-1)/g),m);
   for(ll d=2;d*d<=g;d++)if(g%d==0){
      int sz=0;
      while(g%d==0){g/=d; sz++;}
      a=sub_root(d,sz,a,m);
   }
   if(g>1)a=sub_root(g,1,a,m); return a;
}

int main(){
   int q; cin>>q;
   while(q--){
      ll k,a,m; scanf("%lld%lld%lld",&m,&k,&a);
      printf("%lld\n",mod_root(k,a,m));
   }
   return 0;
}
0