【JZOJ 5809】【NOIP2008模拟】数羊 (数学 逆元)

问题描述
牧羊人A和牧羊人B总是很无聊，所以他们要玩一个游戏。A有a只羊，B有b只羊。他们想要知道a^b的因子和是多少。这就很为难两个牧羊人了，由于答案太大，你能不能告诉我答案取模9901的数。
输入
仅一行，为两个正整数a和b
输出
a^b的因子和对9901的余数。
样例输入
2 3
样例输出
15
算法讨论
对于100%的数据，我们将a进行质因数分解，将每个质因子的指数乘上b，最后的答案就是等比数列的和，但是由于涉及取模，所以我们除法要用逆元。
等比数列求和：
逆元：由费马小定理可知x^(p-2)(mod p)为x的逆元，设c为b的逆元，则a/b=a*c

#include <cstdio>
#include <cmath>
#include <algorithm>
#define MAX_N 50000006
#define ll long long
#define mo 9901
using namespace std;
int a,b,n,m;
ll ans=1;
bool flag,flag1;

ll power(ll a,ll b)
{
    ll ans=1,base=a;
    while (b!=0)
    {
        if (b & 1!=0)
            ans=(ans*base) % mo;
        base=(base*base) % mo;
        b>>=1;
    }
    return ans;
}

int main()
{
    freopen("sheep.in","r",stdin);
    freopen("sheep.out","w",stdout);
    scanf("%d%d",&a,&b);
    int z1,z2,p=2;
    ll xx,c;
    while (a!=1)
    {
        flag1=0; z2=0;
        while (a % p==0 && !flag)
        {
            a/=p;
            z1=p;
            z2++;
            flag1=1;
        }
        if (flag || flag1)
        {
            z2*=b;
            xx=1-power(z1,z2+1);
            c=power(1-z1,mo-2);
            ans=(ans*(xx*c) % mo) % mo;
        }
        if (flag || a==1)
            break;
        p++;
    }
    printf("%lld",ans);
    fclose(stdin); fclose(stdout);
}