模的情況下取log模板

//接口：log(a,b,n) log 以a爲底，b的對數，餘數是n。不存在返回-1
LL log_mod (LL a,LL b,LL n)
{
    LL m,v,e=1,i;
    m=(LL)sqrt(n+0.5);
    v=inv(get_inv(a,m),n);
    map<int,int>x;
    x[1]=0;
    for(i=1; i<m; i++)
    {
        e=(e*a)%n;
        if(!x.count(e))x[e]=i;
    }
    for(i=0; i<m; i++)
    {
        if(x.count(b))return i*m+x[b];
        b=(b*v)%n;
    }
    return -1;
}