判断数字是否为质数

#include<stdio.h>
#include<stdlib.h>

int zhi(int data)
{
if (data <= 1)
{
return 0;
}
else if (data == 2 || data == 3)
{
return 1;
}
else
{
int flag = 1;
for (int 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("不是质数");
}
system("pause");

}