C语言——判断是否是素数

题目：判断一个整数是否是素数，若是返回1，否则返回0。

代码：

#include<stdio.h>
int fun(int m)
{
	int k=2;
	while(k<=m&&(m%k))
		k++;
	if(m==k)
		return 1;
	else 
		return 0;
 } 
 
 int main()
 {
 	int n;
 	printf("\nPlease enter n:");
 	scanf("%d",&n);
 	if(fun(n))
 		printf("YES!\n");
 	else
 		printf("NO!\n");
 }

运行结果：