【PTA L1-028】判断素数 开方判断质数

L1-028 判断素数 （10 分）

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

输入格式：

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

输出格式：

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

输入样例：

2
11
111

输出样例：

Yes
No

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
 
void pri(int n)
{
	if(n==1)
	{
		cout<<"No"<<endl;
		return ;
	}
	int t=sqrt(n);
	for(int i=2;i<=t;i++)
	{
		if(n%i==0)
		{
			cout<<"No"<<endl;
			return ;
		}
	}
	cout<<"Yes"<<endl;
	return ;
}
 
int main()
{
	int n,t;
	cin>>n;
	while(n--)
	{
		cin>>t;
		pri(t);
	}
	return 0;
}