L1-028 判断素数

本题的目标很简单，就是判断一个给定的正整数是否素数。

输入格式：

输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于231的需要判断的正整数。

输出格式：

对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No。

输入样例：

2
11
111

输出样例：

Yes
No

 评价是别忘了1

#include <iostream>
#include<algorithm>
#include<math.h>
using namespace std;
const int maxsize = 10;
int main() {
	int N;
	long a[maxsize+1]={0};
	int flag[maxsize+1] = { 0 };
	cin >> N;
	for (int i = 0; i < N; i++) {
		cin >> a[i];
	}
	for (int i = 0; i < N; i++) {
		for (int j = 2; j < sqrt(a[i]); j++) {
			if (a[i] % j == 0) {
				flag[i] = 1;//不是素数
				break;
			}
		}
	}
	for (int i = 0; i < N; i++) {
		if (a[i] == 1) cout << "No" << endl;
		else {
			if (flag[i] == 1) cout << "No" << endl;
			else cout << "Yes" << endl;
		}
	}
	return 0;
}