L1-028 判斷素數 (10 分)

作者: 陳越

單位: 浙江大學

時間限制: 400 ms

內存限制: 64 MB

代碼長度限制: 16 KB

本題的目標很簡單，就是判斷一個給定的正整數是否素數。

輸入格式：

輸入在第一行給出一個正整數N（≤ 10），隨後N行，每行給出一個小於2​31​​的需要判斷的正整數。

輸出格式：

對每個需要判斷的正整數，如果它是素數，則在一行中輸出Yes，否則輸出No。

輸入樣例：

2
11
111

輸出樣例：

Yes
No

---

源碼：

#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<iostream>
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;
}