素数求和问题

描述
现在给你N个数（0 < N < 1000），现在要求你写出一个程序，找出这N个数中的所有素数，并求和。
输入
第一行给出整数M(0

#include <iostream>#include <cmath>using namespace std;int main(){    int i,n,m,num,prime = 0,sum = 0;    cin >> n;    while(n--)    {        cin >> m;        while(m--)        {            cin>>num;            if(num == 1) prime = 1;            for(i = 2 ; i <= sqrt(num);i++)            {                if(num % i == 0) prime = 1;            }            if(prime == 0) sum += num;            prime = 0;        }            cout << sum << endl;            sum = 0;    }    return 0;}

优化代码：

#include<stdio.h>#include <math.h>int main(){    int m,n,i,j,a[1000],flag=0;    long s;    scanf("%d",&m);    while(m--)    {        s=0;        scanf("%d",&n);        for(i = 0;i < n;i++) scanf("%d",&a[i]);        for(i = 0;i < n;i++)        {            if(a[i]==1) continue;            flag=0;            for(j=2;j<=sqrt(a[i]);j++)            {                if(a[i]%j==0)                {flag=1;break;}                    }            if(flag==0) s+=a[i];        }        printf("%d\n",s);    }    return 0;}