C：判断输入的整数是否为素数

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>

int main()
{
	int prime(int n);
	int n;
	printf("请输入需要判断的整数:\n");
	scanf("%d", &n);
	prime(n);
	return 0;
}
int prime(int n)
{
	int i, a = 0;
	if (n < 2)
	{
		printf("%d不是素数", n);
		return 0;
	}
	for (i = 2; i < n; i++)
	{
		if (n % i == 0)
			a++;
	}
	if (a == 0)
		printf("%d是素数", n);
	else
		printf("%d不是素数", n);
}

以前写的，好像结合了老师上机题的答案。