利用标志flag,设计一个函数，检测一个数是否为质数(素数)

#include<stdio.h>
#include<stdlib.h>
//设计一个函数，检测一个数是否为质数(素数) 
//只能被1和自身整除的数才是质数  
int zhishu(int data)
{
	if(data<=1)//小于等于1都不是素数 
	{
		return 0;
	}
	else if(data==2 || data==3)//这2个可以直接判断是质数 
	{
		return 1;
	}
	else
	{
		int flag=1;//假设是质数(素数)
		for(int i=2;i<data;i++)
		{
			if(data%i==0)
			{
				flag = 0;//不是质数 
				break;
			}
		} 
		return flag;
	}
}
int main()
{
	int data;
	scanf("%d",&data);
	int flag = zhishu(data);
	if(flag)
	{
		printf("%d是质数(素数)",data);
	}
	else
	{
		printf("%d不是质数(素数)",data);
	}
	return 0;
}