Cod sursa(job #2103587)

Utilizator dragos.galeteanu2001Dragos Iulian dragos.galeteanu2001 Data 10 ianuarie 2018 15:16:43
Problema Suma divizorilor Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <fstream>
#define MOD 9901

using namespace std;

ifstream f("sumdiv.in");
ofstream g("sumdiv.out");

long long power(long long x, int p)
{
	long long rez = 1;
	while (p) {
		if (p & 1) rez = (rez*x) % MOD;
		x = (x*x) % MOD;
		p >>= 1;
	}
	return rez;
}

void solve(long long a, long long int b)
{
	long long sum = 1;

	for (int i = 2; i * i <= a; i++) {
		if (a % i) continue;
		int pw = 0;
		while (!a % i) {
			a /= i;
			pw++;
		}
		pw *= b;
		if (!i % MOD) continue;
		if (i % MOD == 1) {
		    sum = (sum*(pw + 1)) % MOD;
		}
		else {
		    sum = (sum*(power(i, pw + 1) - 1)*(power(i - 1, MOD - 2))) % MOD;
		}
	}

	if (a != 1) {
		if (a % MOD == 1) {
		    sum = (sum*(b + 1)) % MOD;
		}
		else if (a % MOD) {
		    sum = (sum*(power(a, b + 1) - 1) * power(a - 1, MOD - 2)) % MOD;
		}
	}

	g << sum << '\n';
}

int main()
{
    int a, b;
	f >> a >> b;
	solve(a, b);
	return 0;
}