【PAT甲级题解记录】1015 Reversible Primes (20 分)

【PAT甲级题解记录】1015 Reversible Primes (20 分)

前言

Problem：1015 Reversible Primes (20 分)

Tags：素数 进制转换

Difficulty：剧情模式 想流点汗 想流点血 死而无憾

Address：1015 Reversible Primes (20 分)

问题描述

求一个数在给定进制下的逆序数和顺序数是否都为素数。

解题思路

写个反转加进制转化，再判断下素数即可，题目数据小无需过多操作。

参考代码

#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;

string Decimal_to(int d, int R) {
    string ans = "";
    while (d) {
        int r = d % R; // 余数
        d /= R;
        ans = to_string(r) + ans;
    }
    return ans;
}

int to_Decimal(string num, int R) {
    int ans = 0;
    for (int i = 0; i < int(num.size()); i++) {
        ans = ans * R + (num[i] - '0');
    }
    return ans;
}

int is_prime(int a) {
    if (a == 1) return 0;
    if (a == 2) return 1;
    for (int i = 2; i < sqrt(a) + 1; i++) {
        if (a % i == 0) {
            return 0;
        }
    }
    return 1;
}

void solution_1015() {
    int N;
    while (cin >> N && N >= 0) {
        int R;
        cin >> R;
        string s = Decimal_to(N, R); // 将N转换为R进制
        reverse(s.begin(), s.end()); // N在R进制下反转
        int d = to_Decimal(s, R); // 反转后的十进制值
        if (is_prime(N) && is_prime(d)) {
            cout << "Yes" << endl;
        } else {
            cout << "No" << endl;
        }

    }

}
int main() {
    solution_1015();
    return 0;
}