POJ1845:Sumdiv(求因子和+逆元+质因子分解)好题

题目链接:http://poj.org/problem?id=1845

定义: 满足a*k≡1 (mod p)的k值就是a关于p的乘法逆元。
为什么要有乘法逆元呢?
当我们要求(a/b) mod p的值,且a很大,无法直接求得a/b的值时,我们就要用到乘法逆元。 我们可以通过求b关于p的乘法逆元k,将a乘上k再模p,
即(a*k) mod p。其结果与(a/b) mod p等价。
 题目解析:让求a^b的因子和modk,因为是大数没法直接求,因为求因子和函数是乘性函数,所以首先要质因子分解,化成n=p1^a1*p2^a2*p3^a3****Ps^as,那么

s(n)=[(p1^a1+1 -1)/(p1-1)]*[(p2^a2+1 -1)/(p2-1)]*[(p3^a3+1 -1)/(p3-1)]***[(ps^as+1 -1)/(ps-1)];(因子和)

又因为s(n)%mod等于每一个部分取模,所以可以逐步求解,如求(p1^a1+1  -1)/(p1-1)%mod,在这里就要运用除法取模所以要用到乘法逆元的概念,

(a/b) %p= ( a *b^(-1)%p) ,又因为(a^b) % p = ((a % p)^b) % p

所以(p1^a1+1  -1)/(p1-1)%mod==(((p1%mod)^a1+1 -1)%mod*(p1-1)^-1)%mod;

当然存在逆元的前提是gcd(a,p)==1;

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#define N  500010
#define mod 9901
typedef __int64 ll;
using namespace std;
ll a,b,X,Y;
ll ans[N],num[N],top;
ll pow(ll x,ll k)
{
    ll t=1;
    while(k)
    {
        if(k&1) t=((t%mod)*(x%mod))%mod;
        k>>=1;
        x=((x%mod)*(x%mod))%mod;
    }
    return t;
}
void extend(__int64 A,__int64 B,__int64 &x1, __int64 &y1)
{
    if(B==0)
    {
        x1=1;
        y1=0;
        return ;
    }
    extend(B,A%B,x1,y1);
    ll t=x1;
    x1=y1;
    y1=t-(A/B)*y1;
    return ;
}
void solve()
{
    ll sum=1,A,xx;
    for(int i=0; i<top; i++)
    {
        if(ans[i]%mod==0) continue;//关键的两个判断,关系到求逆元。 如果ans[i]%mod=0,那么有等级公式可以看出,原式小于0,所以也只能利用原式求,结果为1
        if(ans[i]%mod==1)//即mod|(ans[i]-1),因为ans[i]>=2,所以ans[i]不可能等于1,这是gcd(ans[i]-1,mod)==mod,不存在逆元,无法利用扩展欧几里得求逆元
        {                                  //这时为(1+ans[i]^1+ans[i]^2+.....+ans[i]^num[i])%mod=(num[i]+1)%mod;
            sum=(sum*(num[i]+1))%mod;
            continue;
        }
        A=pow(ans[i],num[i]+1);
        A=(A-1)%mod;
        extend(ans[i]-1,mod,X,Y);//因为ans[i]为素数,ans[i]-1为偶数,所以ans[i]-1与9901互质
        xx=(X%mod+mod)%mod;
        A=((A%mod)*(xx%mod))%mod;
        sum=(sum*A)%mod;
    }
    printf("%I64d\n",sum);
}
int main()
{
    while(scanf("%I64d%I64d",&a,&b)!=EOF)
    {
        if(a==0)
        {
            printf("0\n");
            continue;
        }
        else if(a==1||b==0)
        {
            printf("1\n");
            continue;
        }
        ll t=a;
        top=0;
        memset(num,0,sizeof(num));
        for(int i=2; i*i<=a; i++)
        {
            if(t%i==0)
            {
                num[top]++;
                ans[top]=i;
                t/=i;
                while(t%i==0)
                {
                    num[top]++;
                    t/=i;
                }
                top++;
            }
        }
        if(t>1)
        {
            num[top]++;
            ans[top++]=t;
        }
        for(int i=0; i<top; i++)
        {
            num[i]*=b;
        }
        solve();
    }
    return 0;
}

 

 

posted @ 2015-01-25 16:20  人艰不拆_zmc  阅读(415)  评论(0编辑  收藏  举报