写一个判素数的函数，在主函数输入一个整数，输出是否为素数的信息

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