任意進位制下的可逆素數 C++

題目描述

任意進位制下的可逆素數是這樣定義的：它自身是一個素數，當它轉換為任意進位制之後，把所有數字逆序，得到的新數字的值仍然是一個素數。比如，73在十進位制下是一個素數，它在十進位制下的逆37也是一個素數，那就稱它是十進位制下是一個可逆素數。

輸入格式:
輸入有多行，每行包括兩個正整數，數N以及基R，其中N<100000, 1<D<=10, 遇到負數則退出。

輸出格式:
對每一行輸入，如果是可逆素數，則輸出“Yes”，否則輸出“No”

輸入樣例:
在這裡給出一組輸入。例如：

73 10
23 2
23 10
-2

輸出樣例:
在這裡給出相應的輸出。例如：

Yes
Yes
No

樣例說明：23的二進位制是10111，其逆為11101，對應的十進位制數是29，所以是可逆素數。

參考程式碼

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <iostream>

using namespace std;

int N,D;
int a[100005];

int f()
{
    int sum = 0;
    int i = 1;
    int t2 = 1;
    while(N)
    {
        a[i++] = N%D; 
        N /= D;   
    }
    for(int j = i-1;j>=1;j--)
    {
        sum += a[j]*t2;
        t2 *= D;
    }
    return sum;
} 
bool isprime(int t1)
{
    if(t1==1)
        return false; 
    
    for(int i = 2;i*i<=t1;i++)
    {
        if(t1%i==0)
            return false;
    }
    
    return true;
}
int main()
{
    //freopen("test.in","r",stdin);
    //freopen("test.out","w",stdout);
    
    int t;
    
    while(scanf("%d",&N)==1)
    {
        if(N>0)
            scanf("%d",&D);
        else
            break;
        t = f();
        if(isprime(t))
            cout <<"Yes"<<endl;
        else
            cout <<"No"<<endl;
    }

    return 0;
}