Cod sursa(job #3221838)

Utilizator Radu_BicliBiclineru Radu Radu_Bicli Data 8 aprilie 2024 10:56:51
Problema Suma divizorilor Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("sumdiv.in");
ofstream fout("sumdiv.out");
const int mod = 9901;
int a, b;

static inline int Put(int a, int n) {
    int p = 1;
    while(n) {
        if(n & 1) p = (p * a) % mod;
        a = (a * a) % mod;
        n >>= 1;
    }
    return p;
}

static inline int Inv(int a) {
    return Put(a, mod - 2);
}

static inline int Divi(int a) {
    int d = 2;
    int sum = 1;
    while(d * d <= a) {
        int e = 0;
        while(a % d == 0) e++;

        if(e) {
            sum = 1LL * (sum * (Put(d, e * b + 1) - 1 + mod) % mod) % mod;
            sum = 1LL * (sum * Inv(d - 1)) % mod;
        }

        d++;
    }
    if(1 < a) {
        sum = 1LL * (sum * (Put(a, b + 1) - 1 + mod) % mod) % mod;
        sum = 1LL * (sum * Inv(a - 1)) % mod;
    }

    return sum;
}

int main() {
    fin >> a >> b;
    fout << Divi(a);

    return 0;
}