数学问题 | 1015 进制转换与素数问题

这题很简单，可以说没什么难度。但是我必须记录这道题，因为犯了一个小错误导致卡了一个case。

在编写判断一个数是否为素数的函数中，没有加如下加粗语句：

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

考试、比赛的时候一定要细心。pat考的都是细节。

 

完整代码：

#include <stdio.h>
#include <memory.h>
#include <math.h>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#include <map>

#define I scanf
#define OL puts
#define O printf
#define F(a,b,c) for(a=b;a<c;a++)
#define FF(a,b) for(a=0;a<b;a++)
#define FG(a,b) for(a=b-1;a>=0;a--)
#define LEN 1010
#define MAX (1<<30)-1
#define V vector<int>

typedef long long ll;

using namespace std;

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

int main(){
//    freopen("1015.txt","r",stdin);
    int n,i,r;
    while(scanf("%d",&n),n>0){
        I("%d",&r);
        if(!isPrime(n)){
            puts("No");
            continue;
        }
        vector<int> v;
        while(n){
            v.push_back(n%r);
            n/=r;
        }
        reverse(v.begin(),v.end());
        int sum=0,t=1;
        FF(i,v.size()){
            sum+=t*v[i];
            t*=r;
        }
        puts(isPrime(sum)?"Yes":"No");
    }
    return 0;
}