ZOJ 3758 Singles' Day

先化成十进制数，然后再判断是否是素数就行了。开始写了一个函数用来判断是否是素数，结果就超时了，然后写在里面判断就过了。还有要用long long，因为数据很大的，不然会wa

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <iomanip>
#include <algorithm>
#include <cctype>
#include <stack>
#include <queue>
#include <string>
#include <cstring>
#include <iomanip>
using namespace std;
int main()
{
	long long n,b;
	while(scanf("%lld%lld",&b,&n)!=EOF)
	{
		long long realNum=1;
		for(long long i=1;i<n;i++)
               realNum=realNum*b+1;
      if(realNum==1)
		{
			cout<<"NO\n";
			continue;
		}
	   long long tmp=sqrt(realNum);
	   bool vis=true;
	   for(long long i=2;i<=tmp;++i)
	     {
		   if(realNum%i==0)
			 {
                  vis=false;
                  break;
			 }
	     }
	    printf(vis==true?"YES\n":"NO\n");
	}
	return 0;
}