算法题 扩展欧几里得算法之线性同余方程（Python）

题目

给定n组数据ai,bi,mi，对于每组数求出一个xi，使其满足ai∗xi≡bi(mod mi)，如果无解则输出impossible。

输入格式

第一行包含整数n。

接下来n行，每行包含一组数据ai,bi,mi。

输出格式

输出共n行，每组数据输出一个整数表示一个满足条件的xi，如果无解则输出impossible。

每组数据结果占一行，结果可能不唯一，输出任意一个满足条件的结果均可。

输出答案必须在int范围之内。

数据范围

1≤n≤10^5
1≤ai,bi,mi≤2∗10^9

输入样例：

2
2 3 6
4 3 5

输出样例：

impossible
7

代码

n = int(input())
x, y = 1, 0

def exgcd(a, b):
    global x, y
    if b == 0:
        x, y = 1, 0
        return a
    d = exgcd(b, a % b)
    x, y = y, x
    y -= (a // b) * x
    return d

for _ in range(n):
    a, b, m = map(int, input().split())
    d = exgcd(a, m)
    if b % d != 0: print('impossible')
    else: print(x*(b//d) % m)