PAT (Advanced Level) 1015. Reversible Primes (20)

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<map>
#include<queue>
using namespace std;

int a,b;
int h[100],tot;

bool f(int x)
{
if(x==1) return 0;
for(int i=2;i*i<=x;i++) if(x%i==0) return 0;
return 1;
}

int main()
{
while(~scanf("%d",&a))
{
if(a<0) break; scanf("%d",&b);
if(f(a)==0) printf("No\n");
else if(a==0) printf("No\n");
else
{
int tmp=a; tot=0;
while(tmp) h[tot++]=tmp%b,tmp=tmp/b;
int num=0;
for(int i=0;i<tot;i++)
num=num+h[i]*(int)pow(b,tot-i-1);
if(f(num)==1) printf("Yes\n");
else  printf("No\n");
}
}
return 0;
}