POJ1845 Sumdiv "简单"的逆元 BPM136

我从未想过会WA的这么惨= =

首先显然用分配律可以化为很多等比数列的和的乘积

然后欧拉定理求个逆元

然后....WA了= =

突然发现很多地方乘起来会爆int，赶快换了LL

然后......WA了= =

看了下discuss发现可能没有逆元，在(A-1)%MOD==0时，这时候可以用变换模值：(A/B)%mod=(A%(mod*B))/B%mod

然后……WA了= =

在我怀疑人生的时候发现KSM会爆LL

这细节……

社会社会

/* ***********************************************
Author        :BPM136
Created Time  :2018/7/14 20:23:34
File Name     :1845.cpp
************************************************ */

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<vector>
using namespace std;

typedef long long LL;

const int N = 5e7 + 5;
const int M = 5000000;
const int MOD = 9901;

LL A,B;

LL mul(LL a,LL b,LL p) {
    LL ret=0;
    for(; b; b>>=1) {
        if(b&1) ret=(ret+a)%p;
        a=a*2%p;
    }
    return ret;
}

LL KSM(LL a,LL k,LL mod) {
    LL ret=1;
    while(k) {
        if(k&1) ret=mul(ret,a,mod);
        a=mul(a,a,mod);
        k>>=1;
    }
    return ret;
}

LL pri_calc() {
    LL ret=1;
    for(int i=2;i*i<=A;i++) {
        if(A%i) continue;
        int num=0;
        while(A%i==0) {
            A/=i;
            num++;
        }
        if((i-1)%MOD==0) (ret*=KSM(i,B*num+1,MOD*(i-1))/(i-1))%=MOD;
        else (ret*=( KSM(i,B*num+1,MOD) - 1) * KSM(i-1,MOD-2,MOD) % MOD)%=MOD;
    }
    if(A>1) {
        if((A-1)%MOD==0) (ret*=KSM(A,B+1,MOD*(A-1))/(A-1))%=MOD;
        else (ret*=(KSM(A,B+1,MOD) - 1) * KSM(A-1,MOD-2,MOD) % MOD)%=MOD;
    }
    return (ret%MOD+MOD)%MOD;
}

int main() {
    while(cin>>A>>B) {

        if(A<=1 || B==0) {
            cout<<1<<endl;
            continue;
        }
        /*
           LL tmp=1;
        LL anss=0;
        for(int i=1;i<=B;i++) tmp=tmp*A;
        for(int i=1;i<=tmp;i++) if(tmp%i==0) anss+=i;
        cerr<<anss%MOD<<endl;
        */

        LL ans=pri_calc();
        cout<<ans<<endl;
    }
    return 0;
}