2021牛客寒假算法基础集训营3 A. 模数的世界

链接

https://ac.nowcoder.com/acm/contest/9983/A

题意

已知 $a,b$，$x\equiv a(\mathrm{m}\mathrm{o}\mathrm{d}p)$，$y\equiv b(\mathrm{m}\mathrm{o}\mathrm{d}p)$，求 $gcd(x,y)\mathrm{m}\mathrm{o}\mathrm{d}p$ 的最大值以及 $x,y$ 的取值。

思路

当 $a,b$ 不都为 $0$ 时，最大值必为 $p-1$。

$\{\begin{array}{l}(p-1)k_1\%p=a\\ (p-1)k_2\%p=b\end{array}$ 可推出可行解： $\{\begin{array}{l}{k}_1=t_{1}p-a\\ k_2=t_{2}p-b\end{array}(t1,t2\ge 1)$

要保证 $k_1,k_2$ 互质，令 $k_2=p-b$，$(t_{1}p-a)\%k_2=1⟹t_{1}p+m{k}_2=1+a$。

因为 $k_2,p$ 互质，所以此方程必有解。

代码

#include <bits/stdc++.h>
#define SZ(x) (int)(x).size()
#define ALL(x) (x).begin(),(x).end()
#define PB push_back
#define EB emplace_back
#define MP make_pair
#define FI first
#define SE second
using namespace std;
typedef double DB;
typedef long double LD;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef vector<int> VI;
typedef vector<PII> VPII;
//head
LL exgcd(LL a,LL b,LL &x,LL &y) {
    if(!b) {x=1,y=0;return a;}
    LL ret=exgcd(b,a%b,y,x);
    y-=a/b*x;
    return ret;
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int T;
    cin>>T;
    while(T--) {
        LL a,b,p;
        cin>>a>>b>>p;
        LL k1=p-a,k2=p-b,x,y;
        if(!a&&!b) cout<<0<<' '<<0<<' '<<0<<'\n';
        else {
            exgcd(p,k2,x,y);
            x=(x%k2+k2)%k2;
            if(!x) x+=k2;
            x*=1+a;
            k1=x*p-a;
            cout<<p-1<<' '<<k1*(p-1)<<' '<<k2*(p-1)<<'\n';
        }
    }
    return 0;
}