【C语言训练】素数问题

题目链接：http://115.28.203.224/problem.php?cid=1010&pid=23

题目描述

编写一个程序判断一个数是否为素数

输入

整数

输出

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

样例输入

5

样例输出

1

代码

#include <iostream>
#include <cmath>
using namespace std;
int main() {
	int i, n;
	while (cin>>n) {
		for (i = 2; i <= sqrt(n); i++) {
			if (n % i == 0) {
				cout<<"0"<<endl;
				break;
			} else {
				cout<<"1"<<endl;
				break;
			}
		}
	}
	return 0;
}