PAT 甲级 1015 Reversible Primes （20 分)

Note：
这题蛮简单的，但是我提交了两次才过。还是素数的判断函数不够熟练。
测试点2是小于等于1的情况，需要直接返回false。
测试点3是边界条件，一开始没写等号。

Code：

#include<iostream>
#include<string>

using namespace std;

bool isPrime(int n) {
	if (n <= 1) return false;
	for (int i = 2; i*i <= n; i++)
	{
		if (n%i == 0) return false;
	}
	return true;
}

int reverse(int n,int radix) {
	int rev[20] = { 0 };
	int base = 1;
	int r = 0;
	int i = 0;
	while (n)
	{
		rev[i] = n % radix;
		n /= radix;
		i++;
	}
	for (int j = i - 1; j >= 0; j--)
	{
		r += base * rev[j];
		base *= radix;
	}
	return r;
}

int main() {
	int n, d;
	while (cin >> n && n >= 0)
	{
		cin >> d;
		int r = reverse(n,d);
		if (isPrime(n) && isPrime(r))
			cout << "Yes" << endl;
		else
			cout << "No" << endl;
	}
	return 0;
}