51Nod 1181 质数中的质数（质数筛法）

1181 质数中的质数（质数筛法）
题目来源： Sgu
基准时间限制：1 秒 空间限制：131072 KB 分值: 0 难度：基础题 收藏 关注
如果一个质数，在质数列表中的编号也是质数，那么就称之为质数中的质数。例如：3 5分别是排第2和第3的质数，所以他们是质数中的质数。现在给出一个数N，求>=N的最小的质数中的质数是多少（可以考虑用质数筛法来做）。
Input
输入一个数N(N <= 10^6)
Output
输出>=N的最小的质数中的质数。
Input示例
20
Output示例
31

我的想法蛮简单。首先素数筛，刚学会，哈哈，然后，把素数都存起来。先判断序列号是不是素数，在要求序列号对应的素数比n大，然后输出。脑子有点蒙，最近确实有点累，我要睡啦。任务算是完成了。明天学一下乘法逆元，这个东西关系到好几道题。必须要会呀。附上代码

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
#define qwe 1000000
int a[qwe];
int b[qwe];
void prime(void)
{
    int i, j;
    int t=1;
    int n=qwe;
    a[1]=1;
    for(i=2;i<n;i++)
    {
        if(a[i]==0)
        {
            for(j=i+i;j<n;j+=i)
            {
                a[j]=1;
            }
        }
    }
    for(i=1;i<n;i++)
    {
        if(a[i]==0)
        b[t++]=i;

    }
}

int main()
{
    int i, j, k, l, n, m, t;
    prime();

    scanf("%d",&n);
    for(i=1;;i++)
    {
        if(a[i]==0&&b[i]>=n)
        {
            printf("%d\n",b[i]);
            break;
        }   
    }
    return 0;
}

我其实一直担心数组会不会太大，要求的内存不够用，所幸。就这样。