L1-028. 判断素数 - PAT

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

输入格式：

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

输出格式：

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

输入样例：

2
11
111

输出样例：

Yes
No

通过这道题，我学会了判断素数的几种方法：

方法一：

bool judge(int n){
    if(n==1)return false;
    for(int i=2;i*i<=n;i++){
        if(n%i==0)return false;
    }
    return true;
}

方法二：

素数一定在6的倍数的两侧。eg.5,7,11,13,17,19

但是在6的倍数两侧的数不一定都是素数eg.25……

bool judge(int n){
    if(n==1)return false;
    if(n==2||n==3)return true;
    if(n%6!=1&&n%6!=5)return false;
    int tmp=sqrt(n);
    for(int i=5;i<=tmp;i+=6){//在6的倍数左右的不一定都是素数
        if(n%i==0||n%(i+2)==0)return false;
    }
    return true;
}

代码如下：

#include<iostream>
#include<cstdio>
#include<cmath>

using namespace std;
bool judge(int n){
    if(n==1)return false;
    if(n==2||n==3)return true;
    if(n%6!=1&&n%6!=5)return false;
    int tmp=sqrt(n);
    for(int i=5;i<=tmp;i+=6){//在6的倍数左右的不一定都是素数
        if(n%i==0||n%(i+2)==0)return false;
    }
    return true;
}

int main(){
    int n,m;
    scanf("%d",&n);
    while(n--){
        scanf("%d",&m);
        if(judge(m)){
            printf("Yes\n");
        }
        else printf("No\n");
    }
}