C语言《素数问题》

编写一个程序判断一个数是否为素数，注意1不是素数，2是素数。

Input
整数

Output
1或0(其中1表示此数为素数,0为表示为不是素数)

Sample Input
5
Sample Output
1

#include<stdio.h>
#include<math.h>
int main(void)
{
	int a,b,i;
	a=0;
	scanf("%d",&b);
	for (i=2;i<b;i++)
	{
		if(b%i==0)
		{
			a++;
		}
	}
	//输入的值小于等于1时都不是素数
	if(b<=1){
		a=1;
	}
	if(a==0)
	{
		printf("1");
	}
	else
	{
		printf("0");
	}
	return 0;
}