题目:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=5203
给一个b和一个n,m就是写成b进制的n个1,然后如果m是质数就输出YES,否则NO。
由于b和n最多只有16,而m相当于1+b+b^2+...+b^(n-1) <b^n
就算b和n都取16也不会超过long long;
我的做法是先求出所有的取值:f[b][n];
这里可以递推,f[i][j] = f[i][j-1]*i+1
然后判断下f[b][n]是否是质数保存到m[b][n]中。
处理完之后就可以直接输出结果了。
PS:比赛的时候开始还在担心数字太大,处理会不会太慢,结果没想到预处理几乎不用什么时间,估计是这些数字里质数本来就不多吧。
#include<cstdio>
#include<cmath>
#include<cstring>
long long f[20][20];
bool m[20][20];
int main(){
for(int i=1; i<=16; i++) f[i][1]=1;
for(int i=2; i<=16; i++){
for(int j=2; j<=16; j++){
f[i][j] = f[i][j-1]*i+1;
}
}
memset(m,0,sizeof(m));
for(int i=2; i<=16; i++){
for(int j=1; j<=16; j++){
if(f[i][j]==1) m[i][j]=1;
for(int k=2; k<=(int)(sqrt(f[i][j])); k++){
if(!(f[i][j]%k)){
m[i][j]=1;
break;
}
}
}
}
int a, b;
while(~scanf("%d %d", &a, &b)){
puts(m[a][b]?"NO":"YES");
}
return 0;
}