编写一个素数判断函数，调用该函数时，从主调函数中输入一个整数并输出该数是否是素数的信息！

#include<iostream>
#include<cmath>
using namespace std;
int prime(int a)
{
    int b,c,i;
    b=int(sqrt(a));
    for(i=2;i<=b;i++)
       if(a%i==0)
           break;
    if(i>b)
       c=1;
    else
       c=0;
    return c;
}

int main()
{
    int m,n;
    cout<<"Please enter a number:";
    cin>>m;
    if(m<2)
      cout<<"你输入的数小于2"<<endl;
    if(m==2)
        cout<<m<<" is a prime"<<endl;
    else if(m>2)
        {
        n=prime(m);
        if(n)
           cout<<m<<" is a prime"<<endl;
        else
           cout<<m<<" is not a prime"<<endl;
        }
   system("pause");
   return 0;
}