ZOJ 3758 —— Singles' Day

题目：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;
}