素数求和问题

原题链接（素数求和问题）

时间限制：3000 ms  |  内存限制：65535 KB

难度：2

描述

现在给你N个数（0<N<1000），现在要求你写出一个程序，找出这N个数中的所有素数，并求和。 

输入

第一行给出整数M(0<M<10)代表多少组测试数据
每组测试数据第一行给你N，代表该组测试数据的数量。
接下来的N个数为要测试的数据，每个数小于1000

输出

每组测试数据结果占一行，输出给出的测试数据的所有素数和

样例输入

3
5
1 2 3 4 5
8
11 12 13 14 15 16 17 18
10
21 22 23 24 25 26 27 28 29 30

样例输出

10
41
52

#include<iostream>
#include<cmath>
using namespace std;
int main(){
    int M,N,a,flag,sum;
    cin>>M;
    while(M--){
        cin>>N;
        sum=0;
        for(int i=0;i<N;i++){
            cin>>a;
            flag=0;
            if(a==1)
                continue;
            for(int j=2;j<=sqrt(a);j++)
                if(a%j==0)
                    flag=1;//如果a不是素数，flag标记1
            if(flag==0)
                sum+=a; 
        }
        cout<<sum<<endl;
    }
    return 0;
} 

优化：

//素数打表，时间上更优化 
#include<iostream>
int a[1010];//每个数小于1000，数组开到1010足够
int b[1010];//存放素数
using namespace std;
void prime(){
    for(int i=2;i<=1010;i++)
        b[i]=1;
    b[1]=0;//1不是素数
    for(int i=2;i<=1010;i++)
        for(int j=i+i;j<=1010;j+=i)
            b[j]=0;//不是素数标记0，其余为1是素数 
}
int main(){
    int N,M,sum;
    cin>>M;
    while(M--){
        cin>>N;
        int sum=0;
        prime();
        for(int i=0;i<N;i++){
            cin>>a[i];
            if(b[a[i]]==1)
                sum+=a[i];
        }
        cout<<sum<<endl;    
    }
    return 0;
}