PTA R7-1 判断素数

输入格式：

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

输出格式：

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

输入样例：

2
11
111

输出样例：

Yes
No

代码长度限制    16 KB

时间限制    400 ms

内存限制    64 MB

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
int ABC(int n){
	int i;
	if (n==1) return 0;
	for (i=2;i<=sqrt(n);i++) if(n%i==0) return 0;
	return 1;
}
main(){
    int a,b,c;
	scanf("%d",&a);
	for (b=0;b<a;b++) {scanf("%d",&c); if(ABC(c)) printf("Yes\n"); else printf("No\n");}
}