BSGS算法

最新推荐文章于 2022-09-04 09:51:06 发布

xiehaoxuan

最新推荐文章于 2022-09-04 09:51:06 发布

阅读量1.1k

点赞数

分类专栏：数论与数学方法

本文链接：https://blog.csdn.net/weixin_39872717/article/details/82733517

版权

数论与数学方法专栏收录该内容

11 篇文章 0 订阅

订阅专栏

引入

一道签到题

初步分析

这是洛谷9月份月赛的一道题，作为一名蒟蒻，我当时做得真的是惨兮兮。

首先 $N$ 个 $1$ 可以怎么表示呢？不难发现： $a_n=\frac{10^N-1}{9}$ ，故原式可化为 $\frac{10^N-1}{9}\equiv K$ $(mod$ $m)$ ，即 $10^N-1\equiv 9K(mod$ $m)$ ； $---(1)$

再移项 $10^N\equiv 9K+1(mod$ $m)$ ； $---(2)$

这就是BSGS求解问题的一个标准形式：求最小非负整数 $x$ ，满足 $a^x\equiv b(mod$ $m)$ 。

BSGS算法

首先令 $x=i*t-j$ $(j<t)$ ，且 $t=\left \lceil \sqrt m \right \rceil$ ，其中 $\left \lceil \right \rceil$ 表示向上取整，即取大于某个数的最小整数。 $--(3)$

这样原式就变成了： $a^{i*t-j}\equiv b(mod$ $m)$ ，移项： $a^{i*t}\equiv b*a^j(mod$ $m)$ . $---(1)$

枚举 $j=0,1,2,...,t$ ，把 $b*a^j$ 存入 $hash$ 表；

枚举 $i=1,2,...,t$ ，判断 $a^{i*t}$ 是否在 $hash$ 表内，若在，可以证明满足条件的最小 $i$ 值对应最小 $x$ ，故直接输出 $i*t-j$ . $---(4)$

算法结束。

那么求解引入中的签到题，只需要把 $a$ 看成 $10$ ，把 $x$ 看成 $N$ ，把 $b$ 看成 $9K+1$ 即可，最终求得的 $x$ 即为所求 $N$ 值（当然也涉及到一些其它的东西，比如本题很坑的就是两数相乘会爆 $long$ $long$ ，所以要用快速乘）。 $---(5)$

那么为什么 $t$ 要取 $\left \lceil \sqrt m \right \rceil$ 呢，这其实需要用到费马小定理，本文不加以赘述，有兴趣的同学可以自己尝试证明。

注释

$(1)$ 同余方程左右可以乘以同一个数或是同余的两个数；

$(2)$ 同余方程左右可以加减同一个数或是同余的两个数；

$(3)$ 向上取整有许多方法，对于c++而言，可以用 $ceil(sqrt(m))$ 或者 $int(round(sqrt(m)+0.5)$ ；

$(4)$ 由于 $0<=j<t$ ， $\dpi{100} x_{i+1}-x_{i}=(i+1)*t-j_{i+1}-(i*t-j_i)=t+j_{i+1}-j_{i}>0$ ，故 $i$ 越小， $x$ 越小；

$(5)$ 快速乘是一种求解两数相乘取模，但乘的过程中会爆 $long$ $long$ 的方法。

代码

#include<cstdio>
#include<cmath>
#include<map>
using namespace std;
    long long k,m;
    map<long long,long long>mp;
inline long long multi(long long x,long long y,long long mod) //快速乘
{
    long long tmp=(x*y-(long long)(((long double)x*y+0.5)/mod)*mod);
    if (tmp<0) return tmp+mod; else return tmp;
}
long long quickpower(long long a,int b)
{
    long long t=1;
    while (b>0)
    {
        if ((b&1)==1) t=multi(t,a,m);
        if (b>1) a=multi(a,a,m);
        b=b>>1;
    }
    return t;
}
int main()
{
    mp.clear();
    scanf("%lld%lld",&k,&m);
    long long now=(9*k+1)%m;
    mp[now]=0;
    int mm=ceil(sqrt(m));
    for (int i=1;i<=mm;i++) //预处理哈希表
    {
        now=multi(now,10,m);
        mp[now]=i;
    }
    now=1;
    long long q=quickpower(10,mm);
    for (int i=1;i<=mm;i++)
    {
        now=multi(now,q,m);
        if (mp[now])
        {
            printf("%lld",(((long long)(i)*(long long)(mm)-mp[now])%m+m)%m);
            return 0;
        }
    }
    return 0;
}