POJ 2891 线性同余方程组

SHOHOKUKU

于 2020-11-26 20:26:27 发布

阅读量261

点赞数

分类专栏：数学

本文链接：https://blog.csdn.net/neweryyy/article/details/110203216

版权

数学专栏收录该内容

164 篇文章

订阅专栏

题意

传送门 POJ 2891

题解

线性同余方方程，可以使用拓展欧几里德算法计算逆元求解。对于线性同余方程组，可以通过方程数次拓展欧几里德算法迭代求解。假设方程形式为 $a_i\times x\ = \ b_i(mod\ m_i)$ ，在题目中为 $x\ = \ r_i(mod\ a_i)$ 。设已求解的前 $i$ 个方程的解为 $x\ mod\ m$ ，可以写成 $x+m\times t$ ，将其代入下一个需要求解的方程，则有
$a_i(x+m\times t)\ = \ b_i (mod \ m_i)$
将其写为一般形式
$a_i\times m \times t\ = \ b_i-a_i\times x(mod \ m_i)$ 求解方程，将 $t$ 代入解，就得到了前 $i + 1$ 个方程的解。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
typedef long long ll;
typedef vector<ll> vec;
typedef pair<ll, ll> P;

ll gcd(ll a, ll b)
{
    return b == 0 ? a : gcd(b, a % b);
}

ll extgcd(ll a, ll b, ll &x, ll &y)
{
    ll d = a;
    if (b)
    {
        d = extgcd(b, a % b, y, x);
        y -= (a / b) * x;
    }
    else
        x = 1LL, y = 0LL;
    return d;
}

ll inv(ll a, ll m)
{
    ll x, y;
    extgcd(a, m, x, y);
    return (m + x % m) % m;
}

P linear_congruence(const vec &A, const vec &B, const vec &M)
{
    ll x = 0, m = 1;
    for (int i = 0; i < (int)A.size(); ++i)
    {
        ll a = A[i] * m, b = B[i] - A[i] * x, d = gcd(M[i], a);
        if (b % d != 0)
            return P(-1LL, -1LL);
        ll t = b / d * inv(a / d, M[i] / d) % (M[i] / d);
        x = x + m * t;
        m *= M[i] / d;
    }
    return P((m + x % m) % m, m);
}

int main()
{
    int k;
    while (~scanf("%d", &k))
    {
        vec A(k, 1LL), B(k), M(k);
        for (int i = 0; i < k; ++i)
            scanf("%lld%lld", &M[i], &B[i]);
        P res = linear_congruence(A, B, M);
        printf("%lld\n", res.first);
    }
}