判断一个数是否是素数

#include<stdio.h>
#include<math.h>
int fun(int n)
{
int i = 0;
int k = sqrt(n);
for (i = 2; i <= k; i++)
{
if (n % i == 0)
{
printf("NO!\n");
break;
}
if ((i == k) && (n % i != 0))
printf("YES!\n");
}
}

int main()
{
int m;
printf("请输入一个数：");
scanf_s("%d", &m);
fun(m);
system("pause");
return 0;
}