hdu 5150 Sit sit sit

http://acm.hdu.edu.cn/showproblem.php?pid=5151

直接判断是不是素数，然后再注意1就行。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int n;
int a[1001];
bool f[1001];

void Get_prime()
{
    f[1]=true;
    for(int i=2; i<=1000; i++)
    {
        if(!f[i])
        {
            for(int j=i*i; j<=1000; j+=i)
            {
                f[j]=true;
            }
        }
    }
}

int main()
{
     Get_prime();
     while(scanf("%d",&n)!=EOF)
     {
         memset(a,0,sizeof(a));
         int ans=0;
         for(int i=0; i<n; i++)
         {
             scanf("%d",&a[i]);
             if(a[i]==1) ans+=a[i];
             else
             {
                 if(!f[a[i]])
                 {
                     ans+=a[i];
                 }
             }
         }
         printf("%d\n",ans);
     }
     return 0;
}