【SDOI2013 R1 Day1】随机数生成器

传送门:NKOJ2637
这是一道基础的数学题,然而本蒟蒻WA了好久(果然还是太菜了吗qaq)。
以下是题解:
【SDOI2013 R1 Day1】随机数生成器_第1张图片

注意要特判:
【SDOI2013 R1 Day1】随机数生成器_第2张图片

放上渣代码:
这里写图片描述
我不太明白改成注释里的为何会WA掉,求解答。=-=

/*By F.F.Chopin*/ 
#include
#include
#include
#include
#include
#define LL long long
using namespace std;
mapMMap;
int T;
LL p,A,B,x1,t,c,d;
LL ksm(LL a,LL b){
    LL ans=1;
    while(b){
        if(b&1)ans=ans*a%p;
        b>>=1,a=a*a%p; 
    }
    return ans;
}
void bsgs(LL x,LL y){
    LL m=ceil(sqrt(p-1)),ex=ksm(ksm(x,m),p-2),k=1;
    MMap.clear();
    MMap[1]=0;
    //MMap[1]=m+1;
    for(LL i=1;iif(!MMap.count(k))MMap[k]=i;
        //if(!MMap[k])MMap[k]=i;
    }
    for(LL i=0;iif(MMap.count(y)){
            printf("%lld\n",i*m+MMap[y]+1);
            return;
        }
        y=y*ex%p;
    }
    puts("-1");
    return;
}
bool Special_Judge(){
    if(x1==t)return puts("1"),1;
    if(!A){
        if(B==t)return puts("2"),1;
        else return puts("-1"),1;
    }
    if(A==1){
        if(!B)return puts("-1"),1;
        LL tmp=(B+(t-x1))%p*ksm(B,p-2)%p;
        if(tmp<=0)tmp+=p;
        printf("%lld\n",tmp);
        return 1;
    }
    return 0;
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%lld%lld%lld%lld%lld",&p,&A,&B,&x1,&t);
        if(Special_Judge())continue;//Naive!
        c=ksm((x1*(A-1)%p`
B+p)%p,p-2);
        d=(B+(A-1)*t+p)%p;
        bsgs(A,c*d%p);
    }
    return 0;
}

你可能感兴趣的:(BSGS,数学)