POJ 1845 Sumdiv 题解

chomasx

于 2024-02-15 18:16:12 发布

阅读量1.2k

点赞数 33

分类专栏：题解文章标签：算法

本文链接：https://blog.csdn.net/qq_61755102/article/details/136122124

版权

题解专栏收录该内容

2 篇文章

订阅专栏

文章讲述了如何利用质因数分解和分治法，结合快速幂算法求解A^B对9901取模的问题，强调了时间复杂度的控制和编程技巧，如POJ限制和避免使用自动函数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目传送门

题目大意

求 $A^B~\text{mod }9901$ 的值。

题目解答

先对 $A$ 进行质因数分解，可以得到 $A=p_1^{c_1}p_2^{c_2}...p_n^{c_n}$ 。
那么对于 $A^B$ 而言它的质因数分解应该是 $A=p_1^{c_1B}p_2^{c_2B}...p_n^{c_nB}$ 。
由乘法分配律可知 $A^B$ 全部的因数之和应该是
$\prod_{i=1}^{n}\sum_{j=0}^{c_iB}p_i$
。

注意到每一个 $\Sigma$ 都是一个等比数列，但是 $9900$ 不是质数，所以不能求逆元。

这时候我们就可以引入分治法来解决问题。

假设 $c_iB = k$ ， $\Sigma$ 部分我们用 $s u m (p, k)$ 代替。

当 $k$ 为奇数时，展开 $s u m (p, k)$ 得 $1+p+p^2+p^3+...+p^k)$ 。

提出一个 $p^{\frac{k+1}{2}}$ 得原式 $(1+p+p^2+...+p^{\frac{k-1}{2}})\times{(1+p^{\frac{k+1}{2}})}$

$=sum(p,(k-1)/2)\times{(1+p^{\frac{k+1}{2}})}$ 。

类似的，当 $k$ 为偶数时，原式 $=sum(p,k/2-1)\times(1+p^{\frac{k}{2}})+p^k$ 。

其中幂运算可以用快速幂来解决。时间复杂度应为 $O(A\log^2B)$ 。

技术细节

1.POJ 不支持万能头和 auto。

2.在分解质因数时只能用 $O(\sqrt{n})$ 的复杂度，不可偷懒。

3.建议配合 map 食用质因数分解。

代码实现

//1.NO BITS/STDC++.H
//2.USE THE BEST ALGORITHM YOU KNOW 
//3.NO AUTO
#include <iostream>
#include <cstdio>
#include <map>
using namespace std;
typedef long long ll;
#define P 9901
ll a,b;
map<ll,ll> pri;
ll qpow(ll x,ll y)
{
	ll r=1;
//	x%=P;
	while(y)
	{
		if(y&1) r=r*x%P;
		x=x*x%P,y>>=1;
	}
	return r;
}
ll sigma(ll x,ll y)
{
	if(y==0) return 1;
	if(y&1)
	{
		return /*r=*/((qpow(x,(y+1)>>1)+1)%P*sigma(x,(y-1)>>1)%P);
	}
	else
	{
		return /*r=*/((qpow(x,y>>1)+1)%P*sigma(x,(y>>1)-1)%P+qpow(x,y))%P;
	}
//	cout<<x<<' '<<y<<' '<<r<<endl;
}
int main()
{
	scanf("%lld %lld",&a,&b);
	if(a==0)
	{
		printf("0\n");
		return 0;
	}
	ll sum=1;
	for(ll i=2;i<=a/i;i++)
	{
		while(a%i==0)
			a/=i,++pri[i];
	}
	if(a!=1)
		pri[a]++; 
	for(map<ll,ll>::iterator i=pri.begin();i!=pri.end();i++)
	{
		sum*=sigma(i->first,i->second*b)%P;
		sum%=P;
	}
	printf("%lld\n",sum);
	return 0;
}