扩展欧几里得算法

扩展欧几里得算法

• The extended Euclidean algorithm is an algorithm to compute integers $x$ and $y$ such that $ax+by=gcd(a,b)$ , given $a$ and $b$ .

• 扩欧是对辗转相除法做了一些扩展，缩写 $exgcd()$

• 考虑辗转相除法递推到边界，$b=0,a=gcd$

  • 得到边界的解 $x=1,y=0$

• 反递推上一层的解：

  • 当前层：$b{x}_1+(a\%b)y_1=gcd$
    $$\begin{array}{rl}gcd& =b{x}_1+(a-[\frac{a}{b}]\ast b)y_1\\ & =a{y}_1+b(x_1-[\frac{a}{b}]y_1)\end{array}$$

  • 反推到上一层：$x=y_1,y=x_1-[\frac{a}{b}]y_1)$

• 注意：方程 $ax+by=gcd$ 的解不唯一

  • 设 $x_0,y_0$ 是解，则 $x_0+t\ast b/gcd和y_0-t\ast a/gcd$ 也是解，$t\in N$ （要理解）

P1082 [NOIP2012 提高组] 同余方程

#include <bits/stdc++.h>
#define int long long 
using namespace std;

void exgcd(int a,int b,int &x,int &y)
{
    if(!b) { x=1; y=0; return; } // 边界
    exgcd(b,a%b,x,y); // 当前层
    int t=y; 
    y=x-a/b*y; x=t; // 解出上一层的
}

signed main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int a,b;
    cin>>a>>b;
    int x=0,y=0;
    exgcd(a,b,x,y);
    x=(x%b+b-1)%b+1; // 最小正整数解
    cout<<x<<endl;
    return 0;
}

• 该题的另一种写法：
  • $条件是：ax\equiv 1(modb)，欧拉函数：a^{\phi (b)}\equiv 1(modb)$
  • 再根据同余的传递性：$ax\equiv a^{\phi (b)}(modb)$
  • 解得：$ans=ksm(a,euler(b)-1,b)$
• 能用欧拉函数来解的前提：$a,b$ 互质（题目保证有解，$a,b$ 必然互质）

#include <bits/stdc++.h>
#define int long long 
using namespace std;

int euler(int x)//欧拉函数
{
	int ans=x,sq=sqrt(x*1.0);
	for(int i=2;i<=sq;i++)
	{
		if(x%i==0)
		{
			ans=ans-ans/i;
			while(x%i==0) x/=i;
		}
	}
	if(x>1) ans=ans-ans/x;
	return ans;
}
int ksm(int a,int b,int p)
{
    int ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%p;
        a=a*a%p; b >>= 1;
    }
    return ans;
}

signed main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int a,b;
    cin>>a>>b;
    int ans=ksm(a,euler(b)-1,b);
    cout<<ans<<endl;
	return 0;
}