Cod sursa(job #2090890)

Utilizator rares_ciocieaRares Andrei Ciociea rares_ciociea Data 18 decembrie 2017 20:18:55
Problema Pascal Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.54 kb
#include<fstream>
using namespace std;
ifstream in("pascal.in");
ofstream out("pascal.out");
int v2[5000001],v3[5000001],v5[5000001];
int legendre(int n,int p)
{
    int exp=0,nr=p;
    while(n/nr)
    {
        exp+=n/nr;
        nr*=p;
    }
    return exp;
}
void legendre1(int n,int p,int v[])
{
    int i,exp,ci;
    v[0]=0;
    for(i=1;i<=n;i++)
    {
        v[i]=legendre(i,p);
    }
}
int main()
{
    int r,d,i,j,cnt=0,ok=0;
    in>>r>>d;
    if(d==2||d==4||d==6)
        legendre1(r,2,v2);
    if(d==3||d==6)
        legendre1(r,3,v3);
    if(d==5)
        legendre1(r,5,v5);
    for(i=0;i<r/2;i++)
    {
        if(d==2)
            if(v2[r]-v2[r-i]-v2[i])
                ok++;
        if(d==3)
            if(v3[r]-v3[r-i]-v3[i])
                ok++;
        if(d==4)
            if(v2[r]-v2[r-i]-v2[i]>1)
                ok++;
        if(d==5)
            if(v5[r]-v5[r-i]-v5[i])
                ok++;
        if(d==6)
            if(v2[r]-v2[r-i]-v2[i] && v3[r]-v3[r-i]-v3[i])
                ok++;
    }
    if(r%2==0)
    {
        i=r/2+1;
        if(d==2)
            if(v2[r]-v2[r-i]-v2[i])
                cnt++;
        if(d==3)
            if(v3[r]-v3[r-i]-v3[i])
                cnt++;
        if(d==4)
            if(v2[r]-v2[r-i]-v2[i]>1)
                cnt++;
        if(d==5)
            if(v5[r]-v5[r-i]-v5[i])
                cnt++;
        if(d==6)
            if(v2[r]-v2[r-i]-v2[i] && v3[r]-v3[r-i]-v3[i])
                cnt++;
    }
    out<<cnt*2+ok;
    return 0;
}