#include<stdio.h>
#include<stdlib.h> #include<cmath> typedef unsigned long long LL; LL modular_multi(LL x,LL y,LL mo) { LL t; x%=mo; for(t=0;y;x=(x<<1)%mo,y>>=1) if (y&1) t=(t+x)%mo; return t; } LL modular_exp(LL num,LL t,LL mo) { LL ret=1,temp=num%mo; for(;t;t>>=1,temp=modular_multi(temp,temp,mo)) if (t&1) ret=modular_multi(ret,mo); return ret; } bool miller_rabbin(LL n) { if (n==2)return true; if (n<2||!(n&1))return false; int t=0; LL a,x,y,u=n-1; while((u&1)==0) t++,u>>=1; for(int i=0;i<10;i++) { a=rand()%(n-1)+1; x=modular_exp(a,u,n); for(int j=0;j<t;j++) { y=modular_multi(x,n); if (y==1&&x!=1&&x!=n-1) return false; ///其中用到定理，如果对模n存在1的非平凡平方根，则n是合数。 ///如果一个数x满足方程x^2≡1 (mod n),但x不等于对模n来说1的两个‘平凡’平方根：1或-1，则x是对模n来说1的非平凡平方根 x=y; } if (x!=1)///根据费马小定理,若n是素数，有a^(n-1)≡1(mod n).因此n不可能是素数 return false; } return true; } int main() { int n,m; long long sum,mul; while(scanf("%d%d",&n,&m)!=EOF) { sum=1; for(int i=1;i<=m-1;i++) { mul=1; for(int j=1;j<=i;j++) mul*=n; sum=sum+mul; } if(miller_rabbin(sum)) printf("YES\n"); else printf("NO\n"); } return 0; }
