【PAT】1015 Reversible Primes (20 分)

题目链接

数组

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

int arr[100010];

bool check(int n){
    if(n==1)
        return false;
    for(int i=2;i<=sqrt(n);i++){
        if(n%i==0){
            //cout<<i<<endl;
            return false;
        }
    }
    return true;
}

int main()
{
    int n,d;

    while(true){
        cin>>n;
        vector <int>q;
        if(n>=0){
            cin>>d;
            //这里不能少噢，因为接下来的n被改变了
            //我就说为什么最后一个测试点过不了😓
            if(check(n)==false){
                cout<<"No"<<endl;
                continue;
            }
            //转换为d进制
            int index=0;
            while(n>0){
                int temp=n%d;
                arr[index++]=temp;
                n/=d;
            }
            index--;
            int fan=0;
            for(int j=index;j>=0;j--)
                fan+=arr[j]*pow(d,index-j);

            //将反的再转回十进制
            //cout<<fan<<endl;
            if(check(n)&&check(fan)){
                cout<<"Yes"<<endl;
            }else{
                cout<<"No"<<endl;
            }
        }else{
            break;
        }
    }
    return 0;
}

动态数组

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

bool check(int n){
    if(n==1)
        return false;
    for(int i=2;i<=sqrt(n);i++){
        if(n%i==0){
            //cout<<i<<endl;
            return false;
        }
    }
    return true;
}

int main()
{
    int n,d;

    while(true){
        cin>>n;
        vector <int>q;
        if(n>=0){
            cin>>d;
            if(check(n)==false){
                cout<<"No"<<endl;
                continue;
            }
            //转换为d进制
            while(n>0){
                int temp=n%d;
                q.push_back(temp);
                n/=d;
            }
            int fan=0;
            for(int j=0;j<q.size();j++)
                fan+=q[j]*pow(d,q.size()-j-1);

            //将反的再转回十进制
            //cout<<fan<<endl;
            if(check(n)&&check(fan)){
                cout<<"Yes"<<endl;
            }else{
                cout<<"No"<<endl;
            }
        }else{
            break;
        }
    }
    return 0;
}