PAT 1015. Reversible Primes (20)

只想说 英文不好 理解题意出现偏差 以为只要转换后是素数就行了 没想到输入也要判断是否为素数 测试点3

测试点1 测试转换后为1的情况

#include<vector>
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<map>
#include<stack>
using namespace std;
#define INF 100000000
#define min(a,b) a<b?a:b
int  prim(int n)
{
    int i;
    if(n==1||n==0)
        return 0;
    if(n==2||n==3)
        return 1;
    for(i=2;i<=sqrt(n);i++)
    if(n%i==0)
        return 0;
    return 1;
}
int main()
{
    int n,d;
    char a[100];
    int sum=0;
    while(scanf("%d",&n)!=EOF&&n>=0)
    {

        sum=0;
        int s=n;
        int i=0;
        scanf("%d",&d);

    while(s!=0)
    {
        a[i++]=s%d;
        s/=d;
    }
    for(int j=0;j<i;j++)
    {
        sum=sum*d+a[j];
    }
    if(prim(sum)==1&&prim(n))
        printf("Yes\n");
    else
        printf("No\n");


    }

    return 0;
}