POJ 1845 Sumdiv

最新推荐文章于 2019-03-25 09:53:15 发布

muczzcy

最新推荐文章于 2019-03-25 09:53:15 发布

阅读量664

点赞数

分类专栏：初等数论快速幂素数

本文链接：https://blog.csdn.net/imutzcy/article/details/17734483

版权

初等数论同时被 3 个专栏收录

14 篇文章 0 订阅

订阅专栏

素数

6 篇文章 0 订阅

订阅专栏

快速幂

4 篇文章 0 订阅

订阅专栏

Sol：求A^B所有约数和% MOD的结果。

根据唯一分解定理将A进行因式分解就ok.

等比数列通过奇偶性的判断处理下就行了。

A^B=p1^(a1*B)*p2^(a2*B)*...*pn^(an*B);

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cmath>

using namespace std;

const int MOD = 9901;

const int maxn = 10000;
int prime[maxn+1];
void getPrime()
{
	memset(prime,0,sizeof(prime));
	for(int i=2;i<=maxn;i++)
	{
		if(!prime[i])  prime[++prime[0]]=i;
		for(int j=1;j<=prime[0]&&prime[j]<=maxn/i;j++)
		{
			prime[prime[j]*i]=1;
			if(i%prime[j]==0)  break;
		}
	}
}


long long factor[100][2];
int fatCnt;
int getFactors(long long x)
{
	fatCnt=0;
	long long tmp=x;
	for(int i=1;prime[i]<=tmp/prime[i];i++)
	{
		factor[fatCnt][1]=0;
		if(tmp%prime[i]==0)
		{
			factor[fatCnt][0]=prime[i];
			while(tmp%prime[i]==0)
			{
				factor[fatCnt][1]++;
				tmp/=prime[i];
			}
			fatCnt++;
		}
	}
	if(tmp!=1)
	{
		factor[fatCnt][0]=tmp;
		factor[fatCnt++][1]=1;
	}
	return fatCnt;
}

long long pow_m(long long a,long long n)
{
	long long ret=1;
	long long tmp=a%MOD;
	while(n)
	{
		if(n&1)  ret=(ret*tmp)%MOD;
		tmp=tmp*tmp%MOD;
		n>>=1;
	}
	return ret;
}

//计算a+P+P^2+....+P^N
long long sum(long long p,long long n)
{
	if(p==0) return 0;
	if(n==0) return 1;
	if(n&1)  return ((1+pow_m(p,n/2+1))%MOD*sum(p,n/2)%MOD)%MOD;
	else return ((1+pow_m(p,n/2+1))%MOD*sum(p,n/2-1)+pow_m(p,n/2)%MOD)%MOD;
}

long long solve(long long A,long long B)
{
	getFactors(A);
	long long ans=1;
	for(int i=0;i<fatCnt;i++)
	{
		ans*=(sum(factor[i][0],B*factor[i][1])%MOD);
		ans%=MOD;
	}
	return ans;
}

long long a,b,Ans;

int main()
{
	getPrime();
	while(~scanf("%lld%lld",&a,&b))
	{
		Ans=solve(a,b);
		printf("%lld\n",Ans);
	}
	return 0;
}