Cod sursa(job #2471845)

Utilizator simonalpsLupsa Simona Cristina simonalps Data 11 octombrie 2019 16:58:03
Problema Suma si numarul divizorilor Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.46 kb
#include <iostream>
#include <fstream>
#define mod 9973
using namespace std;
ifstream f("ssnd.in");
ofstream g("ssnd.out");

int c[1000001], pr[1000001], k=0;

void ciur()
{
    int i, j;
    for(i=2; i<=100000; i++)
    {
        if(c[i]==0)
        {
            pr[k++]=i;
            for(j=i+i; j<=1000000; j+=i)
                c[j]=1;
        }
    }
}

int logpow (int b, int e)
{
    int rez=1;
    while(e)
    {
        if(e%2==1)
        {
            e--;
            rez*=b;
            rez%=mod;
        }
        else
        {
            e/=2;
            b*=b;
            b%=mod;
        }
    }
    return rez;
}

void calc(int n, int&ndiv, int&sdiv)
{
    ndiv=1;
    sdiv=1;
    int p=0;
    int i;
    for(i=0; i<k && pr[i]*pr[i]<=n; i++)
    {
        p=0;
        while(n%pr[i]==0)
        {
            n/=pr[i];
            p++;
        }
        if(p)
        {
            ndiv*=(p+1);
            sdiv*=(logpow(pr[i], p+1)-1);
            sdiv%=mod;
            sdiv*=logpow(pr[i]-1, mod-2);
            sdiv%=mod;
        }
    }
    if(n!=1)
    {
        ndiv*=2;
        sdiv*=(n+1);
        sdiv%=mod;
    }
}

void solve(int t)
{
    int n, i;
    int ndiv, sdiv;
    while(t)
    {
        f>>n;
        calc(n, ndiv, sdiv);
        g<<ndiv<<" "<<sdiv<<"\n";
        t--;
    }
}

int main()
{
    int t;
    f>>t;
    ciur();
    solve(t);
    return 0;
}