编写一个函数：判断一个整数是否素数，如果是返回1，否则返回0

#include<iostream>
using namespace std;
int work(int n);
void main()
{
int n,f;
cout<<"input n"<<endl;
cin>>n;
f=work(n);
cout<<f<<endl;
}
int work(int n)
{
int i=2;
if(n==1||n==2) return 1;
while(i<n)
{
if(n%i!=0)
i++;
else
return 0;
}
return 1;
}

#include <math.h>

bool IsPrime(int iNum)
{
int j,k;
if(iNum < 2)
{
return false;
}

k = sqrt((double)iNum);
for(j=2;j<=k;j++)
{
if(iNum%j==0)
{
return false;
}
}

return true;
}

int isPrime(int m)
{
int half=m/2;
for(int i=2;i<=half;i++)
{
if(m%i==0)
return 0;