【POJ 1845】 Sumdiv

【题目链接】

           点击打开链接

【算法】

         不妨先将A分解质因数

         A = p1^q1p2^p2p3^p3..pn^qn

         那么,A^B = p1^q1Bp2^q2B...pn^qnB

         根据约数和定理，A^B的约数和就是 ：

        (p1^0 + p1^1 + .. p1^q1B)(p2^0 + p2^1 + ... p2^q2B) ... (pn^0 + pn^1 + ... + pn^qnB)

        显然可以用等比数列求和来做，注意特判逆元不存在的情况

【代码】

          

#include <algorithm>
#include <bitset>
#include <cctype>
#include <cerrno>
#include <clocale>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <limits>
#include <list>
#include <map>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <utility>
#include <vector>
#include <cwchar>
#include <cwctype>
#include <stack>
#include <limits.h>
using namespace std;
#define P 9901

long long i,len,ans,A,B;

struct info
{
		long long p,q;
} d[100010];

inline long long power(long long a,long long n)
{
		long long ans = 1,b = a;
		while (n)
		{
				if (n & 1) ans = (ans * b) % P;
				b = (b * b) % P;
				n >>= 1;	
		}	
		return ans;
}
inline long long inv(long long x)
{
		return power(x,P-2);
}
inline long long calc(long long a,long long n)
{
		return (power(a,n+1) - a % P + P) % P * inv(a-1) % P;
}

int main() 
{
		
		scanf("%lld%lld",&A,&B);
		for (i = 2; i <= sqrt(A); i++)
		{
				if (A % i == 0)	
				{
						d[++len] = (info){i,0};
						while (A % i == 0)
						{
								A /= i;
								d[len].q++;
						}
				}
		}
		if (A > 1) d[++len] = (info){A,1};
		for (i = 1; i <= len; i++) d[i].q *= B;
		ans = 1;
		for (i = 1; i <= len; i++)
		{
				if ((d[i].p - 1) % P == 0) ans = (ans * (d[i].q + 1)) % P;
				else ans = (ans * (calc(d[i].p,d[i].q) + 1)) % P;
		}
		printf("%lld\n",ans);
					
		return 0;
	
}