ZOJ 3758 —— Singles' Day

来源:互联网 发布:网络泡妞骨灰级教程 编辑:程序博客网 时间:2024/06/05 17:43

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


0 0