結果

問題 No.981 一般冪乗根
ユーザー nonamaenonamae
提出日時 2021-11-19 04:07:10
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 10 ms / 6,000 ms
コード長 2,513 bytes
コンパイル時間 991 ms
コンパイル使用メモリ 96,292 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-09 05:43:17
合計ジャッジ時間 36,229 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 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 3 ms
5,376 KB
testcase_16 AC 4 ms
5,376 KB
testcase_17 AC 3 ms
5,376 KB
testcase_18 AC 3 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 5 ms
5,376 KB
testcase_26 AC 4 ms
5,376 KB
testcase_27 AC 3 ms
5,376 KB
testcase_28 AC 10 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) {
    if(k>0&&a%p==0)return 0;
    k%=p-1;
    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 k,a,p;
        scanf("%lld%lld%lld",&p, &k, &a);
        printf("%lld\n",kth_root(a, k, p));
    }
}
0