洛谷P1516 青蛙的约会【扩展欧几里得算法】

题目链接：P1516 青蛙的约会

程序说明：

问题转化为求解线性同余方程：x + k * m ≡ y + k * n (mod L)

方程等价于：x + k * m = z * L + y + k * n

合并同类项，化简后得：k * (n − m) + z * L = x - y （将负号化为正号方便处理）

设 a = n - m，b = L，c = x - y，即求线性方程：a * k + b * z = c 的解，扩展欧几里得求出整数解k，再扩大 c / gcd(a, b)倍即为方程的通解。

需要注意的地方有两点：n - m有可能为负数，需特判当n - m为负数时方程两边同时乘负一将负数化为正数（gcd的参数为正整数）；最后的答案要处理一下，所求答案是最小的整数解，模除 L / gcd(a, b)即可得到最小整数解，另外避免出现负数可以加上一个L / gcd(a, b)再模L / gcd(a, b)，这样可以保证所得到的整数解是正数。

代码如下：

#include <iostream>
using namespace std;
typedef long long LL;
int x, y, m, n, l;

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

int main() {
    cin>>x>>y>>m>>n>>l;
    int a = n - m, b = l, c = x - y;
    //exgcd的参数为正数，负数化正
    if(a < 0) {
        a = -a;
        c = -c;
    }
    int x1, y1;
    //求解方程：a * x + b * y = c
    int d = exgcd(a, b, x1, y1);
    int mod = l / d;
    //gcd(a, b)|c时方程才有整数解
    if(c % d) 
        cout<<"Impossible"<<endl;
    else 
        cout<<((LL)x1 * c / d % mod + mod) % mod; //转换为long long类型避免第一个乘法溢出
    return 0;
}