c語言：實現一個函數，判斷一個數是不是素數。

#include <stdio.h>
#include <math.h>
int prime(int num) //prime表示素數
{
int i = 0;
int k = 0;
k = sqrt(num);
for (i = 2; i <= k; i++)
{
if (num%i == 0)
{
   return 0;
}
}
return 1;
}
int main()
{
int num;
int ret;
printf("請輸入一個大於1的正整數：");
scanf("%d", &num);
ret = prime(num);
if (ret == 1)
{
printf("%d是素數\n", num);
}
else
{
printf("%d不是素數\n", num);
}
return 0;
}