洛谷 P5656 【模板】二元一次不定方程(exgcd)

给一个二元一次不定方程

$$ax+by=c$$

• 无整数解：输出 -1
• 有整数解：
  • 有正整数解：输出解的个数，输出 $x,y$ 的最小值和最大值
  • 无正整数解：输出 $x,y$ 的最小正整数值

思路：

首先是欧几里得算法，用来求 $\mathrm{g}\mathrm{c}\mathrm{d}(a,b)$，首先假设 $a>b$，假如 $b\mid a$，则 $\mathrm{g}\mathrm{c}\mathrm{d}(a,b)=b$，假如 $b\nmid a$，则：

$$a=kb+c$$

令 $d\mid a,d\mid b$，则显然有 $d\mid c$ ，令 $d=\mathrm{g}\mathrm{c}\mathrm{d}(a,b)$，则 $\mathrm{g}\mathrm{c}\mathrm{d}(a,b)\mid c$，而 $c=a\mathrm{m}\mathrm{o}\mathrm{d}b$，因此 $\mathrm{g}\mathrm{c}\mathrm{d}(a,b)=\mathrm{g}\mathrm{c}\mathrm{d}(b,a\mathrm{m}\mathrm{o}\mathrm{d}b)$。

int gcd(int a,int b){
	if(!b)return a;
	return gcd(b,a%b);
}

然后是扩展欧几里得，用来求 $ax+by=\mathrm{g}\mathrm{c}\mathrm{d}(a,b)$ 的一组可行解，首先令

$$\begin{array}{rl}a{x}_1+b{y}_1& =\mathrm{g}\mathrm{c}\mathrm{d}(a,b)\\ b{x}_2+(a\mathrm{m}\mathrm{o}\mathrm{d}b)y_2& =\mathrm{g}\mathrm{c}\mathrm{d}(b,a\mathrm{m}\mathrm{o}\mathrm{d}b)\end{array}$$

根据欧几里得算法，有 $\mathrm{g}\mathrm{c}\mathrm{d}(a,b)=\mathrm{g}\mathrm{c}\mathrm{d}(b,a\mathrm{m}\mathrm{o}\mathrm{d}b)$，因此：

$$\begin{array}{rl}a{x}_1+b{y}_1& =b{x}_2+(a\mathrm{m}\mathrm{o}\mathrm{d}b)y_2\\ a{x}_1+b{y}_1& =b{x}_2+(a-b\lfloor \frac{a}{b}\rfloor )y_2\\ a{x}_1+b{y}_1& =a{y}_2+b(x_2-\lfloor \frac{a}{b}\rfloor y_2)\end{array}$$

可得：

$$\begin{array}{rl}{x}_1& =y_2\\ y_1& =x_2-\lfloor \frac{a}{b}\rfloor y_2\end{array}$$

通过递归可解：

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

由此可以解出 $x_0,y_0$ 使得

$$a{x}_0+b{y}_0=\mathrm{g}\mathrm{c}\mathrm{d}(a,b)$$

成立，为了解出 $ax+by=c$ 的一个特解，对上式做一些变换，得：

$$a\frac{x_{0}c}{\mathrm{g}\mathrm{c}\mathrm{d}(a,b)}+b\frac{y_{0}c}{\mathrm{g}\mathrm{c}\mathrm{d}(a,b)}=c$$

因此可得原方程得一个特解 $x_1,y_1$：

$$x_1=\frac{x_{0}c}{\mathrm{g}\mathrm{c}\mathrm{d}(a,b)},y_1=\frac{y_{0}c}{\mathrm{g}\mathrm{c}\mathrm{d}(a,b)}$$

然后构造原方程通解，令 $d\in \mathbb{Q}$，则

$$a(x_1+db)+b(y_1-da)=c$$

并且 $da,db\in \mathbb{Z}$，易知 $d=\frac{1}{\mathrm{g}\mathrm{c}\mathrm{d}(a,b)}$，令

$$d_x=\frac{b}{\mathrm{g}\mathrm{c}\mathrm{d}(a,b)},d_y=\frac{a}{\mathrm{g}\mathrm{c}\mathrm{d}(a,b)}$$

则通解为：

$$\begin{array}{rl}x& =x_1+s{d}_x\\ y& =y_1-s{d}_y\end{array}$$

其中 $s\in \mathbb{Z}$。

考虑解出来的 $x,y$ 均为正整数，则

$$\begin{array}{rl}{x}_1+s{d}_x>0& ⟹s>-\frac{x_1}{d_x}\\ y_1-s{d}_y>0& ⟹s<\frac{y_1}{d_y}\end{array}$$

因此

$$L=\lceil \frac{-x_1+1}{d_x}\rceil \le s\le \lfloor \frac{y_1-1}{d_y}\rfloor =R$$

假如 $L>R$，则不存在正整数解，$s=L$ 时得 $x_{\min }$，$s=R$ 时得 $x_{\max }$；

否则，存在正整数解，解的个数是 $R-L+1$，$s=L$ 时可以得到 $x_{\min },y_{\max }$ ，$s=R$ 时可以得到 $x_{\max },y_{\min }$

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
typedef long long ll;
int T;
ll a,b,c;
ll exgcd(ll a,ll b,ll &x,ll &y){
    if(!b){x=1,y=0;return a;}
    ll d=exgcd(b,a%b,x,y);
    ll t=x;x=y;y=t-a/b*y;
    return d;
}
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    scanf("%d",&T);
    while(T--){
        ll x0,y0;
        scanf("%lld%lld%lld",&a,&b,&c);
        ll d=exgcd(a,b,x0,y0);
        if(c%d){
            printf("-1\n");
            continue;
        }
        ll x1=x0*c/d,y1=y0*c/d;
        ll dx=b/d,dy=a/d;
        ll L=(ll)ceil((-x1+1)*1.0/dx);
        ll R=(ll)floor((y1-1)*1.0/dy);
        ll xmin=x1+L*dx;
        ll ymin=y1-R*dy;
        if(L>R){
            printf("%lld %lld\n",xmin,ymin);
            continue;
        }
        ll xmax=x1+R*dx,ymax=y1-L*dy;
        printf("%lld %lld %lld %lld %lld\n",R-L+1,xmin,ymin,xmax,ymax);
    }
    return 0;
}