PTA 1015 Reversible Primes

题目描述

分析：将一个数在k进制下翻转，判断这个数和翻转后的数是否都是素数，埃氏筛找出素数表，完后进制转换

#include<cstdio>
#include<algorithm>
using namespace std;
const int INF=1e5;

bool prime[INF];

void findPrime(){
    fill(prime+2,prime+INF,true);
    for(int i=2;i<INF;i++){
        if(prime[i]){
            for(int j=i+i;j<INF;j+=i){
                prime[j]=false;
            }
        }
    }
}

int change(int n,int d){
    int k=0;
    do{
        k=k*d+n%d;
        n/=d;
    }while(n!=0);
    return k;
}
int main(){
    findPrime();
    int n,k;
    scanf("%d",&n);
    while(n>=0){
        scanf("%d",&k);
        if(!prime[n]){
            printf("No\n");
        }
        else{
          if(prime[change(n,k)]){
              printf("Yes\n");
          }
          else{
              printf("No\n");
          }
        }
        scanf("%d",&n);
    }
}