高效素数筛法|北航复试上机题|C++

输入一个整数n(2<=n<=10000)，要求输出所有从1到这个整数之间(不包括1和这个整数)个位为1的素数，如果没有则输出-1。

输入描述:

输入有多组数据。
每组一行，输入n。

输出描述:

输出所有从1到这个整数之间(不包括1和这个整数)个位为1的素数(素数之间用空格隔开，最后一个素数后面没有空格)，如果没有则输出-1。

示例1

输入
100
输出
11 31 41 61 71

题解

题目（见图1）两点要求：1.筛出素数 2.个位为1

素数（质数）的定义很简单，如果一个数只能被1和它本身整除，就是素数。

如果暴力遍历所有输入，算法复杂度为O(n*n)，下面用的素数筛法复杂度则为O(nloglogn)

素数筛法核心思想：
从2开始，初始化所有输入为（isPrime=True），并将每个素数的倍数都标记为合数（isPrime=False），筛出素数；小技巧是当前素数i的倍数标记从ii开始，而不是2i（已经在前面标记过），再次减少运行时间。

程序流程：
1.走一遍上述素数筛法，将筛出的素数存入vector中
2.对10求模，判断所有素数中个位是否为1，按要求输出

Tips：为什么选用vector存储素数，而不是数组？
数组创建时需要明确大小，向量则相当于变长数组，也可以通过下标来访问元素，动态变化的特性使其规避了浪费内存或访问越界的问题。只需掌握如何存取即可

C++11

#include<iostream>
#include<cstdio>
#include<cmath>
#include<vector>
 
using namespace std;
 
const int MAX=10001;
 
vector<int> prime; //存储数组
bool isPrime[MAX]; //True为素数
 
void Initial(){
    for(int i=0;i<MAX;i++){
        isPrime[i]=true;
    }
    isPrime[0]=false;
    isPrime[1]=false;  //0 1不是素数
    for(int i=2;i<MAX;i++){
        if(!isPrime[i]){
            continue;
        }
        prime.push_back(i);
        for(int j=i*i;j<MAX;j=j+i){ //从i*i开始标记
            isPrime[j]=false;
        }
    }
    return ;
}
 
int main(){
    Initial();
    int n;
    while(cin>>n){
        int count=0;
        for(int i=0;i<prime.size()&&prime[i]<n;i++){ //判断上一步有没有素数，没有输出-1
            count++;
        }
        if(count==0)
            cout<<"-1"<<endl;
        else{
            for(int i=0;i<prime.size()&&prime[i]<n;i++){
                if(prime[i]%10==1) //求模筛选个位为1的素数
                    cout<<prime[i]<<" ";
            }
            cout<<endl;
        }
    }
    return 0;
}