Cod sursa(job #1195330)

Utilizator bciobanuBogdan Ciobanu bciobanu Data 6 iunie 2014 21:34:22
Problema Ciurul lui Eratosthenes Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.71 kb
#include <cstdio>
#include <cmath>
using namespace std;
const int NMAX=1000;
bool p[NMAX];
int atkin(int n)
{
    int limit=ceil(sqrt(n>>2)),secventa[]={2,4},index,k1,k,i,j,i2,nr=0;
    for(i=1;i<limit;++i)
    {
        index=0;
        k1=(i*i)<<2;
        j=1;
        if(!(i%3))
        {
            while(1)
            {
                k=k1+j*j;
                if(k>n) break;
                *(p+k)=!*(p+k);
                j+=*(secventa+(++index&1));
            }
        }
        else
        {
            while(1)
            {
                k=k1+j*j;
                if(k>n) break;
                *(p+k)=!*(p+k);
                j+=2;
            }
        }
    }
    limit=ceil(sqrt(n/3));
    for(i=1;i<limit;i+=2)
    {
        index=1;
        k1=3*i*i;
        j=2;
        while(1)
        {
            k=k1+j*j;
            if(k>n) break;
            *(p+k)=!*(p+k);
            j+=*(secventa+(++index&1));
        }
    }
    limit=ceil(sqrt(n));
    for(i=1;i<limit;++i)
    {
        k1=3*i*i;
        if(!(i&1)) j=1,index=0;
        else j=2,index=1;
        while(j<i)
        {
            k=k1-j*j;
            if(k<n) *(p+k)=!*(p+k);
            j+=*(secventa+(++index&1));
        }
    }
    *(p+2)=*(p+3)=1;
    for(i=5;i<=limit;++i)
    {
        if(*(p+i))
        {
            i2=i*i;
            for(j=i2;j<=n;j+=i2) *(p+j)=0;
        }
    }
    for(i=2;i<=n;++i)
    {
        if(*(p+i)) ++nr;
    }
    return nr;
}
int main()
{
    freopen("ciur.in","r",stdin);
    freopen("ciur.out","w",stdout);
    int n;
    scanf("%d",&n);
    printf("%d\n",atkin(n));
    fclose(stdin);
    fclose(stdout);
    return 0;
}