2023.1.16[模板]BSGS/exBSGS

2023.1.16 [模板]BSGS/exBSGS

全称Boy Step Girl Step

给定一个质数 p，以及一个整数 a，一个整数 b，现在要求你计算一个最小的非负整数 l,

满足\(a^x \equiv b (mod p)\)

算法流程

设t = \(\lceil \sqrt p \rceil\) ,x = i * t - m (0\(\leq\)i\(\leq\)t , 0\(\leq\)m\(\lt\)t - 1)

\(a^{i * t - b} \equiv b\) (mod p)

\(({a^t})^i \equiv b * a^m\) (mod p)

我们建立一个map,将b * \(a^0\) ~ b * \(a^{t - 1}\) 全部推进去，将值与指数建立映射关系，然后再枚举i，递推算出左式，判断map中是否有值，如果有值则返回答案ans = i * t - map[val]

时间复杂度O(\(\sqrt p\))

inline int BSGS(int a,int b,int p)//a ^ x = b (% p)
{
	map <int,int> q;
	int t = sqrt(p);
	for(int i = 0;i <= t - 1;i++)
		q[(1ll * ksm(a,i,p) * b % p)] = i;
	int c = ksm(a,t,p);
	for(int i = 1;i <= t;i++)
	{
		int now = ksm(c,i,p);
		if(q.find(now) != q.end())
		{
			int x = q[now];
			return i * t - x;
		}
	}
	return -1;
}

exBSGS

那么如果a，p不互质呢？

设d = gcd(a,p);

对于一个同余方程\(a^x \equiv b\) (mod p)

我们可以把它转化成二元一次不定方程的形式 \(a^x\) + py = b

给a分离系数得到 a\(a^{x-1}\) + py = b

想办法消去d，\(\frac ad a^{x-1} + \frac pd y = \frac bd\)

多次迭代，直到d = 1，可以得到以下柿子

\(\frac a{d_1d_2...d_k} a^{x-k} + \frac p{d_1d_2...d_k} y = \frac b{d_1d_2...d_k}\)

转化为原式 \(\frac a{d_1d_2...d_k} a^{x-k} \equiv \frac b{d_1d_2...d_k}\) \(mod (\frac p{d_1d_2...d_k})\)

记录一个变量r = \(\frac a{d_1d_2...d_k} a^{x-k}\)，每次迭代时累乘上\(\frac a{d_i}\)

把r移向同余号右侧，那么我们就得到了一个标准的BSGS，但是同余运算中不能左右同除，于是我们用exgcd算出r关于\(\frac p{d_1d_2...d_k}\)的逆元，然后两边同乘以逆元，在跑一边BSGS即可

Code

#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
typedef long long ll;
using namespace std;
inline int ksm(ll base,int pts,int mod)
{
	ll ans = 1;
	for(;pts > 0;pts >>= 1,base = 1ll * base * base % mod)
		if(pts & 1)
			ans = ans * base % mod;
	return ans;
}
inline int BSGS(int a,ll b,int p)//a ^ x = b (% p)
{
	__gnu_pbds::gp_hash_table <ll,int> q;
	b %= p;
	int t = ceil(sqrt(p));
	for(int i = 0;i <= t;i++)
		q[(ksm(a,i,p) * b % p)] = i;
	ll c = ksm(a,t,p),now = 1;
	for(int i = 1;i <= t;i++)
	{
		now = now * c % p;
		if(q.find(now) != q.end())
		{
			int x = q[now];
			return (i * t - x + p) % p;
		}
	}
	return -1;
}
inline int exgcd(ll &x,int &y,int a,int b)
{
	if(b == 0)
	{
		x = 1;
		y = 0;
		return a;
	}
	int ret = exgcd(x,y,b,a % b);
	int k = y;
	y = x - (a / b) * y;
	x = k;
	return ret;
}
inline int gcd(int a,int b)
{
	if(b == 0) return a;
	return gcd(b,a % b);
}
ll ot = 1;
inline int exBSGS(int a,int b,int p)
{
	if(ot == b) return 0;
	int x,y;
	int d = gcd(a,p);
	if(d == 1)
	{
		ll inv;
		exgcd(inv,y,ot,p);
		inv = ((inv % p) + p) % p;
		return BSGS(a,(1ll * inv * b) % p,p);
	}
	if(b % d != 0) return -1;
	ot = ot * (a / d) % (p / d);
	int ret = exBSGS(a,b / d,p / d);
	if(ret == -1) return -1;
	return ret + 1;
}
signed main()
{
	int a,p,b;
	scanf("%d%d%d",&a,&p,&b);
	while(a || p || b)
	{
		a %= p;b %= p;
		if(b == 1 || p == 1)
		{
			puts("0");
			scanf("%d%d%d",&a,&p,&b);
			continue;
		}
		ot = 1;
		int k = exBSGS(a,b,p);
		if(k == -1)
			printf("No Solution\n");
		else
			printf("%lld\n",k);
		scanf("%d%d%d",&a,&p,&b);
	}
	return 0;
}

语法tip:

C++中有一种比map 和 unordered_map更快的hash映射gp_hash_table，用法和普通map相同

头文件

#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>

命名空间

using namespace __gnu_pbds

定义hash

(__gnu_pbds::)gp_hash_table <int,int> q;