Cod sursa(job #2834437)

Utilizator in-the-loopElliot Anderson in-the-loop Data 16 ianuarie 2022 23:19:23
Problema Suma si numarul divizorilor Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.16 kb
#include <bits/stdc++.h>
#define LL long long
#define SQRT 1000000
#define MOD 9973

using namespace std;

ifstream fin("ssnd.in");
ofstream fout("ssnd.out");

int prime[SQRT + 2];

void generare()
{
    bitset<SQRT + 1> ciur;
    ciur[0] = 1;
    ciur[1] = 1;
    prime[0] = 0;
    for (int i = 2; i <= SQRT; i++)
	if (!ciur[i])
	{
	    prime[++prime[0]] = i;
	    for (int j = i * 2; j <= SQRT; j += i)
		ciur[j] = 1;
	}
}

int calcNrDiv(LL n)
{
    int nrDiv = 1;
    for (int i = 1; i <= prime[0]; i++)
    {
	int p = 0;
	while (n % prime[i] == 0)
	{
	    p++;
	    n /= prime[i];
	}

	nrDiv *= p + 1;
    }

    if (n > 1)
	nrDiv *= 2;
    return nrDiv;
}

LL calcSumDiv(LL n)
{
    LL sumDiv = 1;
    for (int i = 1; i <= prime[0]; i++)
    {
	LL pwr = prime[i];
	while (n % prime[i] == 0)
	{
	    pwr *= prime[i];
	    n /= prime[i];
	}
	
	sumDiv = sumDiv * ((pwr - 1) / (prime[i] - 1)) % MOD;
    }

    if (n > 1)
	sumDiv = sumDiv * ((n * n - 1) / (n - 1)) % MOD;

    return sumDiv % MOD;
}

int main()
{
    int t; 
    generare();
    fin >> t;
    for (int i = 0; i < t; i++)
    {
	LL x;
	fin >> x;
	fout << calcNrDiv(x) << ' ' << calcSumDiv(x) << '\n';
    }
    return 0;
}