LOJ#10211. Sumdiv【乘法逆元】

题目描述 https://loj.ac/problem/10211
约数和公式：对于已经分解的整数A=(p1^k1)*(p2k2)*(p3^k3)*…*(pnkn)
有A的所有因子之和为

分两种情况讨论，p1是9901的倍数，p1不是9901的倍数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
const int modd=9901;
int m,prime[20],c[20],ans=1,a,b;
void devide(int x)
{
	for(int i=2;i<=sqrt(x);i++)
	  if(x%i==0)
	  {
	  	 prime[++m]=i;
	  	 while(x%i==0) x/=i,c[m]++;
	  }
	if(x>1) prime[++m]=x,c[m]=1;
}
int f(ll b,ll p)
{
	ll s=1;
	while(p!=0)
	{
		if(p%2==1) s=s*b%modd;
		b=b*b%modd;
		p/=2;
	}
	return s;
}
int main() 
{ 
   scanf("%d%d",&a,&b);
   devide(a);
   for(int i=1;i<=m;i++)
   {
      if((prime[i]-1)%modd==0) ans=(b*c[i]+1)%modd*ans%modd;
	  else
	  {
	     int x=f(prime[i],b*c[i]+1);	
	     x=(x-1+modd)%modd;
	     int y=f(prime[i]-1,modd-2);
	     ans=ans*x%modd*y%modd;
	  } 	
   } 
   printf("%d",ans); 	
 	
  return 0;	
}