求下列方程的一组解:
ax + by = gcd(a, b)
当gcd(a, b) = 1,即a, b互质的时候,这个方程的解实际上就对应了a关于模b的逆元。
给出伪代码
Input: a, b
Output: a solution to ax + by = gcd(a, b)
1: function extended gcd(a,b)
2: if b = 0 then
3: return (1, 0)
4: else
5: (x, y) ⇐ extended gcd(b, a mod b)
6: return (y, x − (a div b) × y)
7: end if
下面是c++的代码
typedef long long ll;
typedef pair <ll,ll> pll;
#define mp make_pair
pll exgcd(pll res)
{
pll ans,cnt;
if(res.second==0) return mp(1LL,0LL);
else
{
cnt=mp(res.second,res.first%res.second);
ans=exgcd(cnt);
return mp(ans.second,ans.first-(res.first/res.second)*ans.second);
}
}
void slove(ll a,ll b)
{
pll k=mp(a,b);
k=exgcd(k);
cout<<k.first<<" "<<k.second<<endl;
}
下边做个简要分析,这是个修改版的辗转相除算法,是可以算出最后的gcd(a, b)的,考
虑第3行,当当前情况是a = gcd(a, b), b = 0时,则很容易看出x = 1, y = 0是一组解,所以
在第5行的时候我们可以假定x × b + y × (a mod b) = gcd(a, b)。对上式展开可得:
gcd(a, b) = xb + y × (a − (a div b) × b)
= xb + ya − y × (a div b) × b
= y × a + (x − (a div b) × y) × b
所以这个算法在正向计算出gcd(a, b)之后,返回时就计算出了解,而复杂度显然和原先
的一样。