实现一个函数，判断一个数是不是素数。

#include <stdio.h>
#include <math.h>
int prime(int num)
{
int i=0;
int k=0;
int flag;
k=sqrt(num);
flag=1;
for(i=2;i<=k;i++)
{
if(num%i==0)
{
flag=0;
}
if(flag==0)
{
return 0;
}
}
return 1;
}
int main()
{
int num;
int ret;
scanf("%d",&num);
ret=prime(num);
if(ret==1)
{
printf("%d is a sushu;",num);
}
else
{
printf("%d is not a sushu;",num);
}
return 0;
}