NYOJ 22 素数求和问题

来源：http://acm.nyist.net/JudgeOnline/problem.php?pid=22

素数求和问题

时间限制： 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 <stdio.h>
#include <math.h>
int main(void)
{
	int  prime(int a);
	int n,m,i,j,a,s;
	scanf("%d",&n);
	for(i=0;i<n;i++)
	{
		scanf("%d",&m);
		for(j=0,s=0;j<m;j++)
		{
			scanf("%d",&a);
			if(prime(a)==0)
				s=s+a;
		}
		printf("%d\n",s);
	}
	return 0;
	 
}
int  prime(int a)
{
	int j;
	double m;
	m=sqrt(a);
	if(a==1)
		return 1;
	for(j=2;j<m;j++)
	{
		if(a%j==0)
			return 1;
	}
	if(j>m)
		return 0;
}

最优代码：

#include <iostream>
#include<math.h>
using namespace std;
bool vis[1010];
int main()
{
	int n=1010;
	int m =sqrt(n+0.5);
	int c=0;
	for(int i =2;i<=m;i++)
		if(!vis[i])
		{
			for (int j = i*i;j<=n;j+=i)
				vis[j]=1;
		}
		int a;
		cin>>a;
		vis[1]=true;
		while(a--)
		{
			int b;
			cin>>b;
			int sum=0;
			for (int i=0;i<b;i++)
			{
				int x;
				cin>>x;
				if(!vis[x]) sum +=x;
			}
			cout<<sum<<endl;
		}
		return 0;
}