PTA ch5作业 7-1 判断素数

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

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

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

输入样例：
2
11
111
输出样例：
Yes
No

代码：

include

include

include

include

using namespace std;

int ss(int n);

int main(){
int n;
scanf("%d", &n);
int a = (int)malloc(sizeof(int) * n);
for(int i=0; i<n; i++){
scanf("%d", &a[i]);
}
for(int i=0; i<n; i++){
if(ss(a[i])) printf("Yes");
else printf("No");
if(i < n-1) printf("\n");
}
}
int ss(int n){
if(n<2) return 0;
else if(n<4) return 1;
int x = sqrt(n);
for(int i=2;i<x; i++){
if(n%i == 0) return 0;
}
return 1;
}