用函数判断n是否为素数

编写函数ss(n), 判断n是否为素数，是返回1，否返回0。
编写main函数，输入一个数num，调用ss(num)函数，并输出判断结果。

#include<stdio.h>
#include<math.h>
int ss(int n)
{
	int i=2;
		while(i<n)
	{
		if(n%i==0)
		break;
		i++;
	}
	
	return i==n;
}
main()
{
	int num,f;
	printf("输入一个数:");
	scanf("%d",&num);
	f=ss(num);
	if(f==1)
	printf("%d是素数",num);
	else if(f==0)
	printf("%d不是素数",num);				
}

本人小白，还请各位多多关照