牛客网 素数（素数筛法、北航机试）

题目描述

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

输出描述:

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

输入

100

输出

11 31 41 61 71

Solution

素数筛法。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 10000 + 111;
bool prime[maxn];
void get_prime()
{
    memset(prime, true, sizeof(prime));
    prime[0] = prime[1] = false;
    for (int i = 2; i < maxn; i++)
    {
        if (prime[i])
            for (int j = i + i; j < maxn; j += i)
                prime[j] = false;
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    get_prime();
    int n;
    while (~scanf("%d", &n))
    {                                          //flag控制输出空格
        bool flag = false, have_prime = false; //have_prime判断是否有素数
        for (int i = 2; i < n; i++)
        {
            if (prime[i] && (i % 10 == 1))
            {
                have_prime = true;
                if (flag)
                    printf(" ");
                flag = true;
                printf("%d", i);
            }
        }
        if (!have_prime)
            printf("-1\n");
    }
    return 0;
}