51 nod 质数中的质数

好讨厌这种题，又不得不练。。。

#include<cstdio>
#include<cstring>
#include<iostream>
#define long long ll
using namespace std;
const int maxn=1001005;
int p[maxn];      //第 i 个质数
bool pp[maxn];   //记录是不是质数

int get_prime(){
    int pnum=0;
    for(int i=2;i<=maxn;i++){
        if(!pp[i])
            p[pnum++]=i;
        for(int j=0;j<pnum&&p[j]*i<=maxn;j++){
            pp[p[j]*i]=1;
            if(i%p[j]==0)
                break;
        }
    }
    pp[0]=1;
    pp[1]=1;
    return pnum;
}
int main(){
    int cnt=get_prime();
    int n;
    scanf("%d",&n);
    int flag=0;
    for(int i=1;i<maxn;i++){
        if(p[i]==n){
            flag=0;
            for(int j=2;j<i;j++)
                if((i+1)%j==0)
                    flag=1;
            if(!flag){
                printf("%d",p[i]);
                break;
            }
        }
        else if(p[i]>n){
            flag=0;
            for(int j=2;j<i;j++)
                if((i+1)%j==0)
                    flag=1;
            if(!flag){
                printf("%d",p[i]);
                break;
            }
        }
    }
    return 0;
}