C语言函数 判断质数

#include <stdio.h>
int zhi(int data)
{
	if(data<=1)
	{
		return 0;
	}
	else if(data==2)
	{
		return 1;
	}
	else
{

int flag=1;
int i;
for(i=2;i<data;i++)
{
	if(data%i==0)
	{
		flag=0;
		break;
	}
}
return flag;
}
}
void main()  //主函数
{
	int num;
	scanf("%d",&num);  //输入任意一个数
	int flag=zhi(num);
	if(flag==1)
	{
		printf("是质数");
	}
	else
	{
		printf("不是质数");
	}
	
}