【算法学习笔记】35:扩展欧几里得算法求解线性同余方程

线性同余方程问题

线程同余方程问题是指 a x ≡ b ( m o d m ) ax \equiv b~(mod~m) ax≡b (mod m),给定 a a a、 b b b和 m m m,找到一个整数 x x x使得该方程成立,即使得 a x m o d m = b ax~mod~m=b ax mod m=b,随便返回任何一个解都可以。

例如 4 x ≡ 3 ( m o d 5 ) 4x \equiv 3~(mod~5) 4x≡3 (mod 5),那么 x x x的一个可能的解可以是 2 2 2。

接下来用扩展欧几里得算法尝试构造这个解。从 a x ≡ b ( m o d m ) ax \equiv b~(mod~m) ax≡b (mod m)可知,一定存在一个 y y y使得:
a ⋅ x = m ⋅ y + b a \cdot x = m \cdot y + b a⋅x=m⋅y+b

也就是说,因为 a x ax ax模 m m m的余数是 b b b,所以 a x ax ax一定可以表示成 m m m的整数 y y y倍再加上一个 b b b。也就是:
a x − m y = b ax - my = b ax−my=b

令 y ′ = y y' = y y′=y,那么就是:
a x + m y ′ = b ax + my' = b ax+my′=b

因此原线性同余方程问题求 x x x有解,等价于这个方程求 x x x和 y ′ y' y′有解。而根据扩展欧几里得算法里所讨论的, a a a是 g c d ( a , m ) gcd(a,~m) gcd(a, m)的倍数, m m m也是 g c d ( a , m ) gcd(a,~m) gcd(a, m)的倍数,所以它们拼到一起也必须是 g c d ( a , m ) gcd(a,~m) gcd(a, m)的倍数。

因此,这个方程有解的充要条件 是 b b b必须是 g c d ( a , m ) gcd(a,~m) gcd(a, m)的倍数,也即 g c d ( a , m ) ∣ b gcd(a,~m)~|~b gcd(a, m) ∣ b 。

例题:AcWing 878. 线性同余方程

这题最终结果要限制在int范围内,因为 m m m也是在int范围内的,并且:
a x + m y = b ⇔ a ( k m + r ) + m y = b ⇔ a r + m ( a k + y ) = b ax + my =b \\ \Leftrightarrow a(km + r) + my = b \\ \Leftrightarrow ar + m(ak + y) = b ax+my=b⇔a(km+r)+my=b⇔ar+m(ak+y)=b

也就是说,把系数 x x x变成 r = x m o d m r = x~mod~m r=x mod m时,另一个系数只要从 y y y变成 a k + y ak+y ak+y就可以了,其中 k = ⌊ x m ⌋ k = \lfloor \frac{x}{m} \rfloor k=⌊mx⌋。

所以可以直接把结果 x x x模 m m m,一定也是一个合法的解,并且满足在int范围内的要求。

cpp 复制代码
#include <iostream>

using namespace std;

typedef long long LL;

int exgcd(int a, int b, int& x, int& y) {
    if (!b) {
        x = 1, y = 0;
        return a;
    }
    int d = exgcd(b, a % b, y, x);
    // d = b * y + (a % b) * x = b * y + (a - a / b * b) * x
    //   = a * x + b * (y - a / b * x)
    y -= a / b * x;
    return d;
}

int main() {
    int t; cin >> t;
    
    while (t -- ) {
        int a, b, m; cin >> a >> b >> m;
        // ax % m = b, ax + my' = b, iff gcd(a, m) = d | b
        int x, y;
        int d = exgcd(a, m, x, y);
        if (b % d) puts("impossible");
        else cout << (LL)x * (b / d) % m << endl;
    }
    
    return 0;
}
相关推荐
是店小二呀9 分钟前
【动态规划 | 子序列问题】子序列问题的最优解:动态规划方法详解
算法·动态规划·代理模式
啊我不会诶44 分钟前
BD202402跑步 线性求逆元 素数筛 数学
学习·算法·补题
遇见尚硅谷1 小时前
挑战指针题
c语言·数据结构·算法
是阿建吖!1 小时前
【优选算法】分治--归并排序
数据结构·算法·排序算法
CoovallyAIHub1 小时前
轻量?智能?协同?你选的标注工具,到底有没有帮你提效?
深度学习·算法·计算机视觉
roman_日积跬步-终至千里2 小时前
【机器学习】(算法优化一)集成学习之:装袋算法(Bagging):装袋决策树、随机森林、极端随机树
算法·机器学习·集成学习
ikkkkkkkl2 小时前
LeetCode:209.长度最小的子数组&&3.无重复字符的最长子串
数据结构·c++·算法·leetcode
记录Java学习的三木2 小时前
每日一leetcode:移动零
算法
桦02 小时前
[找出字符串中第一个匹配项的下标]
算法·leetcode
恣艺2 小时前
LeetCode 140:单词拆分 II
算法·leetcode·职场和发展