统计十个数中所有素数的和

#include<iostream.h>
#include<math.h>
int prime(int x)
{
int f=1,k;
if (x==1) f=0;
for (k=2; k<=sqrt(x); k++)
if (x%k==0)
{
f=0;
break;
}
return f;
}
void main()
{
int a[10],i,sum=0;
cout<<"输入十个数:"<<endl;
for(i=0;i<10;i++)
{
cin>>a[i];
if (prime (a[i]))
sum+=a[i];
}
cout<<sum<<endl;
}