同余问题学习笔记

同余问题学习笔记

——by sunzz3183

---

扩展欧几里得（exgcd）

命题

对于两个正整数 $a,b$，存在两个整数 $x,y$ 使得

$$ax+by=gcd(a,b)$$

证明

在欧几里得算法的最后一步，即 $b=0$ 时，显然 $x=1,y=0$，使得 $a\times 1+0\times 0=gcd(a,0)$。

若 $b>0$ ,则，$gcd(a,b)=gcd(b,a\mathrm{mod}b)$。假设存在一对整数 $x,y$，满足

$$bx+(a\mathrm{mod}b)y=gcd(b,a\mathrm{mod}b)$$

因为

$$bx+(a\mathrm{mod}b)y=bx+(a-b\lfloor \frac{a}{b}\rfloor )y=ay+b(x-\lfloor \frac{a}{b}\rfloor y)$$

所以令 $x^{\prime }=y,y^{\prime }=x-\lfloor \frac{a}{b}\rfloor y$，就得到了 $a{x}^{\prime }+b{y}^{\prime }=gcd(a,b)$

证毕

代码

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

通式

则对于任意一个

$$a{x}_0+b{y}_0=c$$

当 $gcd(a,b)|c$ 时，等式成立。

则此时

$$x_0=x\frac{c}{\gcd (a,b)},y_0=y\frac{c}{\gcd (a,b)}$$

通解

设二元一次不定方程

$$ax+by=c$$

有一组整数解 $x=x_0,y=y_0$，则一切整数解可以表示成

$$x=x_0-\frac{bt}{\gcd (a,b)},y=y0+\frac{at}{\gcd (a,b)}$$

其中 $t=0,\pm 1,\pm 2,\pm 3,\cdots $

证明

既然 $x0,y0$ 为整数解，当然满足 $a{x}_0+b{y}_0=c$，因此

$$a(x_0-\frac{bt}{\gcd (a,b)})+b(y0+\frac{at}{\gcd (a,b)})=a{x}_0+b{y}_0=c$$

设 $x^{\prime },y^{\prime }$ 为任意一组整数解，则有 $a{x}^{\prime }+b{y}^{\prime }=c$，减去 $a{x}_0+b{y}_0=c$，即得

$$a(x^{\prime }-x_0)+b(y^{\prime }-y_0)=0$$

$$a(x^{\prime }-x_0)=-b(y^{\prime }-y_0)$$

所以我们得到

$$a|(y^{\prime }-y_0)$$

即

$$y^{\prime }=y_0+at$$

将

$$y^{\prime }=y_0+at$$

代入

$$a(x^{\prime }-x_0)+b(y^{\prime }-y_0)=0$$

即得

$$x^{\prime }=x_0-bt$$

因此

$$x^{\prime },y^{\prime }$$

可表示成 $x^{\prime }=x_0-\frac{bt}{\gcd (a,b)},y^{\prime }=y0+\frac{at}{\gcd (a,b)}$ 的形式

所以

$$x=x_0-\frac{bt}{\gcd (a,b)},y=y0+\frac{at}{\gcd (a,b)}$$

为 $ax+by=c$ 的一切整数解

证毕

中国剩余定理（CRT）

命题

已知同余式组

$$\{\begin{array}{c}x\equiv b_1(\mathrm{mod}{m}_1),\\ x\equiv b_2(\mathrm{mod}{m}_2),\\ \cdots ,\\ x\equiv b_n(\mathrm{mod}{m}_n)\end{array}$$

其中 $b\in \mathbb{N},m\in {\mathbb{N}}^{\ast },m_i\perp m_j$

求 $x$

求法

令

$$M=\prod _{i=1}^n{m}_i=m_1{M}_1=m_2{M}_2=\cdots =m_n{M}_n$$

则同时满足同余式组

$$\{\begin{array}{c}x\equiv b_1(\mathrm{mod}{m}_1),\\ x\equiv b_2(\mathrm{mod}{m}_2),\\ \cdots ,\\ x\equiv b_n(\mathrm{mod}{m}_n)\end{array}$$

的正整数解是

$$x\equiv \sum _{i=1}^n{b}_i{M}_i^{\prime }{M}_i(\mathrm{mod}M)$$

这里的 $M_i^{\prime }$ 是满足同余式

$$M_i^{\prime }{M}_i\equiv 1(\mathrm{mod}m)$$

的正整数解

证明

因为 $m_i\perp m_j,M_i=\frac{M}{m_i}$，所以 $m_i\perp M_i$，所以

$$gcd(m_i,M_i)=gcd(m_j,M_j)=1$$

由扩展欧几里得可得，存在两个整数 $M_i^{\prime },n_i$，使得

$$M_i^{\prime }{M}_i+m_i+n_i=1$$

所以存在一个 $M_i^{\prime }$，使得

$$M_i^{\prime }{M}_i\equiv 1(\mathrm{mod}{m}_i)$$

又由当 $i\ne j$时，则由 $m_i\perp m_j,M_j=\frac{M}{m_j}$ 得到 $m_i|M_j$，所以

$$b_j{M}_j^{\prime }{M}_j\equiv 0(\mathrm{mod}{m}_i)$$

所以

$$\sum _{i=1}^n{b}_i{M}_i^{\prime }{M}_i\equiv b_i{M}_i^{\prime }{M}_i\equiv b_i(\mathrm{mod}{m}_i)$$

设 $y$ 为同余式组的一个整数解，则得

$$\{\begin{array}{c}x\equiv y(\mathrm{mod}{m}_1),\\ x\equiv y(\mathrm{mod}{m}_2),\\ \cdots ,\\ x\equiv y(\mathrm{mod}{m}_n)\end{array}$$

也就是

$$\begin{array}{c}{m}_1|(x-y),\\ m_2|(x-y),\\ \cdots ,\\ m_n|(x-y)\end{array}$$

又因为 $m_i\perp m_j$，所以有 $M|(x-y)$，也就是

$$x\equiv y(\mathrm{mod}M)$$

所以

$$x\equiv \sum _{i=1}^n{b}_i{M}_i^{\prime }{M}_i(\mathrm{mod}M)$$

是满足同余式组的唯一正整数解

证毕

代码

inline int exgcd(int a,int b,int &x,int &y){
    if(!b){x=1;y=0;return a;}
    int d=exgcd(b,a%b,y,x);
    y-=(a/b)*x;
    return d;
}
int CRT(int A[],int B[],int k){
    int x,y,a=0,m,n=1;
    for(int i=1;i<=k;i++)n*=A[i];
    for(int i=1;i<=k;i++){
        m=n/A[i];
        exgcd(A[i],m,x,y);
        (a+=y*m*B[i])%=n;
    }
    return (a+n)%n;
}

扩展中国剩余定理（exCRT）

命题

已知同余式组

$$\{\begin{array}{c}X\equiv b_1(\mathrm{mod}{m}_1),\\ X\equiv b_2(\mathrm{mod}{m}_2),\\ \cdots ,\\ X\equiv b_n(\mathrm{mod}{m}_n)\end{array}$$

其中 $b\in \mathbb{N},m\in {\mathbb{N}}^{\ast }$

求 $X$

与 $CRT$ 不同的是，这个 $m_i,m_j$，不一定两两互质。

求法

（建议配合代码食用）

考虑合并

$$\{\begin{array}{c}X\equiv b_1(\mathrm{mod}{m}_1),\\ X\equiv b_2(\mathrm{mod}{m}_2)\end{array}$$
可以写成

$$\{\begin{array}{c}X=b_1+m_{1}x,\\ X=b_2+m_{2}y\end{array}$$

所以

$$b_1+m_{1}x=b_2+m_{2}y$$

整理，得

$$m_{1}x+m_2(-y)=b_2-b_1$$

通过 $exgcd$ 可以得到一组解 $x_0,y_0$

所以

$$m_1{x}_0\frac{b_2-b_1}{\gcd (m_1,m_2)}+m_2{y}_0\frac{b_2-b_1}{\gcd (m_1,m_2)}=\gcd (m_1,m_2)\frac{b_2-b_1}{\gcd (m_1,m_2)}=b_2-b_1$$

因为 $X=m_{1}x+b_1$，现在要最小化 $X$，也就是最小化 $x$，即最小化 $x=x_0\frac{b_2-b_1}{\gcd (m_1,m_2)}$

我们假设 $b_2,m_2$，为已经算出的前 $k$ 对数的答案

于是就可以将这两个同余方程合并为

$$X\equiv m_{1}x+b_1(\mathrm{mod}lcm(m_1,m_2))$$

代码

inline int exgcd(int a,int b,int &x,int &y){
    if(!b){x=1;y=0;return a;}
    int d=exgcd(b,a%b,y,x);
    y-=(a/b)*x;
    return d;
}
int exCRT(int *A,int *B,int k){
    int x,y,a=A[1],ans=B[1];
    for(int i=2;i<=k;i++){
    	int b=A[i],c=((B[i]-ans)%b+b)%b,d=exgcd(a,b,x,y);
    	if(c%d)return -1;
    	x=(__int128)x*(__int128)c/(__int128)d%(b/d);
    	ans+=a*x;
    	a=a/d*b;
    	ans=(ans%a+a)%a;
    }
    return (ans%a+a)%a;
}