判断素数

原题链接：https://www.patest.cn/contests/gplt/L1-028

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

输入格式：

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

输出格式：

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

输入样例：

2
11
111

输出样例：

Yes
No

我的思路：

若n存在除1和它本身之外的因数，那么这个数一定小于等于根号n，则依次判断是否存在整数整除n

需要注意的是：因为n的范围是小于2的31次，所以要用long来存储n

我的代码：

#include <bits/stdc++.h>
using namespace std;
int main(){
	int n;
	long data;
	cin>>n;
	while(n--){
		cin>>data;
		int flag=1;
		long len=sqrt(data);
		if(data == 0 || data == 1)
		  cout<<"No"<<endl;
		else{
		for(int i=2;i<=len;i++)
		 if(data%i == 0){
		 	flag=0;
		 	break;
		 }
		 if(flag == 0)
		   cout<<"No"<<endl;
		 else
		   cout<<"Yes"<<endl;	
		}
	
	} 
	return 0;
}