【代码超详解】洛谷 P3846 [TJOI2007]可爱的质数（BSGS 算法求解离散对数 · 模板）

最新推荐文章于 2026-06-29 10:37:48 发布

原创最新推荐文章于 2026-06-29 10:37:48 发布 · 249 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#详解

ACM-ICPC 专栏收录该内容

238 篇文章

订阅专栏

本文详细解析了洛谷P3846题目的内容，重点讲解了如何利用Berlekamp-Welch-Godunov-Schmidt(BSGS)算法来解决离散对数问题。文章分为三个部分，包括题目描述、算法分析及AC代码实现，为读者提供了清晰的思路和完整的解决方案。

一、题目描述

在这里插入图片描述

二、算法分析说明与代码编写指导

在这里插入图片描述

三、AC 代码

#include<cstdio>
#include<unordered_map>
#include<cmath>
#pragma warning(disable:4996)
using namespace std;
unsigned long long a, b, p, x;
template<typename _Ty> inline _Ty PowerMod(_Ty radix, _Ty exp, const _Ty& mod) {
	_Ty ans = 1; radix %= mod;
	while (exp) {
		if (exp & 1)ans = (ans * radix) % mod;
		exp >>= 1, radix = (radix * radix) % mod;
	}
	return ans % mod;
}
//Solve pow(a, x) ≡ b (% p) for x, GCD(a, p) = 1
template<class _Ty> inline _Ty bsgs(const _Ty& a, const _Ty& b, const _Ty& p) {
	static unordered_map<_Ty, _Ty> u; pair<_Ty, _Ty> q(b % p, 0); _Ty m = ceil(sqrt(p)), L = 1 % p;
	const _Ty fk = a % p, fi = PowerMod(a, m, p); static typename unordered_map<_Ty, _Ty>::iterator I;
	u.clear(), u.emplace(q);
	for (q.second = 1; q.second <= m; ++q.second) { q.first = q.first * fk % p, u.emplace(q); }
	for (_Ty i = 1; i <= m; ++i) { L = L * fi % p, I = u.find(L); if (I != u.end())return i * m - I->second; }
	return -1;
}
int main() {
	scanf("%llu%llu%llu", &p, &a, &b);
	x = bsgs(a, b, p);
	if (x != -1)printf("%llu\n", x);
	else puts("no solution");
	return 0;
}