7-4 判断素数

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

输入格式：

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

输出格式：

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

输入样例：

2
11
111

 输出样例：

Yes
No

#include <iostream>
#include<math.h>

using namespace std;

int main() {
	int N, x, flag;
	cin >> N;
	while (N--)
	{
		flag = 0;
		cin >> x;
		if (x > 1)
			flag = 1;
		if ( x == 1 || (x % 2 == 0 && x!= 2))
	     flag = 0;
		for( int i = 3;i<sqrt(x);i+=2 )
	{
		if( x % i == 0)
		{
			flag = 0;
			break;
		}
	}
		if (flag)
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}