团体程序设计天梯赛-练习集——L1-028 判断素数(平方根)

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

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

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

输入样例：

2
11
111

输出样例：

Yes
No

大写加粗的注意事项：

1. 题目要求需要输入的数是正整数，所以就要考虑1的情况，而1不是素数。
2. 判断一个数是素数时，只需从2遍历到这个数的平方根即可。只要出现一个被整除的数即可退出循环。这样就可以降低时间复杂度，不然会超时。
3. 改进了下，如果这个数能被2整除，直接退出循环，就不用再被偶数整除了。除数直接从3开始，然后+=2。

平方根的使用：

1. cpp中，需要引用头文件

 #include<cmath>
 sqrt(a);

2. java中，Math.sqrt() 方法用于返回参数的算术平方根

double sqrt(double d)

以下是代码：

//cpp
#include<iostream>
#include<cmath>
using namespace std;

int main(){
	int N ;
	cin>>N;
	int a[10];
	for(int i = 0; i < N; i++){
		cin>>a[i];
	}
    
	for(int i = 0; i < N; i++){
		int flag = 0;
        if (a[i] == 1 ) {
            flag = 1;
			cout<<"No"<<endl;
        }else{
			for(int j = 3; j < sqrt(a[i]); j+=2){
				if(a[i] % 2 == 0){
					flag = 1;
					break;
				} 
				if(a[i] % j == 0){
					flag = 1;
					break;
				} 
			}
			if(flag == 0){
				cout<<"Yes"<<endl;
			} else {
				cout<<"No"<<endl;
			}
		}
	}
	return 0;
}

//Java
import java.util.Scanner;

public class Main{
	public static void main(String[] args){
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int[] a = new int[10];
		for(int i = 0; i < N; i++){
			a[i] = sc.nextInt();
		}
		for(int i = 0; i < N; i++){
			int flag = 0;
			if (a[i] == 1 ) {
				flag = 1;
				System.out.println("No");
			}else{
				for(int j = 3; j < Math.sqrt(a[i]); j+=2){
					if(a[i] % 2 == 0){
						flag = 1;
						break;
					} 
					if(a[i] % j == 0){
						flag = 1;
					break;
					} 
				}
				if(flag == 0){
					System.out.println("Yes");
				} else {
					System.out.println("No");
				}
			}
		}
	}
}