浙大PAT甲级-1015

prime：素数；

1不是素数也不是合数。

#include <iostream>#include <vector>using namespace std;int num;bool isPrime(int a){    if(a==1)        return false;    for(int i=2;i<a;i++)        if(a%i==0)            return false;    return true;}void change(int n, int d, vector<int> &x){    int cnt=0;    while(n>0){        x.push_back(n%d);        n/=d;        cnt++;    }    num=cnt;}int Back(int d,vector<int> &x){    int n=0, m=1;    for(int i=num-1;i>=0;i--){        n+=x[i]*m;        m*=d;    }    return n;}int main(){    int n,d, cnt=0;    vector<int> ans;    while(cin>>n){        if(n<0)            break;        cin>>d;        cnt++;        vector<int> x;        if(!isPrime(n)){            ans.push_back(0);            continue;        }        change(n,d,x);        if(!isPrime(Back(d,x))){            ans.push_back(0);            continue;        }        ans.push_back(1);    }    for(int i=0;i<cnt;i++)        if(ans[i]==1)            cout<<"Yes"<<endl;        else            cout<<"No"<<endl;    return 0;}