函数调用判断素数

问题描述：
写一个判断素数的函数，在主函数输入一个整数，输出是否是素数的消息。
输入：一个数
输出：“prime” or “not prime”.
示例 ：
Input ： 97
output ： prime

/*C语言 函数调用判断素数*/
#include<stdio.h>
int judge(int);
int main(void)
{
	int num;
	
	scanf("%d",&num);
	if(judge(num))               //传递实参
	printf("prime\n");
	else
	printf("not prime\n");
	return 0;
}
int judge(int n)     //定义形参
{
	for(int i=2;i<n;i++)      
	  if(n%i==0)            //素数仅可被1和本身整除
	    return 0;
	  else
	    return 1; 
}