POJ 1061 BZOJ 1477 Luogu P1516 青蛙的約會 (擴展歐幾裏得算法)
手動博客搬家: 本文發表於20180226 23:35:26, 原地址https://blog.csdn.net/suncongbo/article/details/79382991
題目鏈接: (poj)http://poj.org/problem?id=1061
(bzoj)http://www.lydsy.com/JudgeOnline/problem.php?id=1477
(Luogu)https://www.luogu.org/problemnew/show/P1516
數據強度對比: 在以上三個OJ中,本題Luogu數據最強。使用一種錯誤代碼在BZOJ與POJ均能AC,而Luogu無法AC.
題目大意:
求解方程\[u+mx\equiv v+nx (\mod p)\]
思路分析:
解同余方程?很經典的使用exgcd算法的問題。(簡單一點的exgcd解同余方程的題目可參照luogu P1082 NOIP 2012 D2 T1 同余方程,題目鏈接https://www.luogu.org/problem/show?pid=1082)
一般來說,如果是形如\(ax\equiv c(\mod b)\)的同余方程都可化為\(ax+by=c\)的形式,用exgcd算法求解後\(x\)的值即為原方程的解。
所以直接化一化式子即可: \[u+mx\equiv v+nx(\mod p)\]\[u+mx-v-nx\equiv 0(\mod p)\]
註意討論正數與負數的情況。現假設\(m>n\).
如果\(\gcd(m-n,p)\)不整除\(|v-u|\)(註意v不一定大於u), 則無解
否則直接exgcd即可。求出\[(m-n)x+py=gcd(m-n,p)\]的一組解,乘以\(\frac{v-u}{gcd(m-n,p)}\)(註意不加絕對值)即可. 於是我們求出了特解。
如何求x>0且最小的解呢? 我們發現若\(ax+by=c\)
註意此處“在數學上”\(A \mod B\)是指\(A\equiv X (\mod B)\)且\(0\le X\lt b\)的唯一的X, 但是在C++語言編程中不能這樣取模,C++中負數取模的含義是
(-A) % B == -(A % B) (A>0,B>0)
例如
(-6) % 5 = -1
(-7) % 4 = -3
(-18) % 9 = 0
其返回值\(x\)滿足\(-B\lt x\le 0\)
因此在數學上負整數\(-A\)對正整數\(B\)取模,就相當於在C++語言中的
(((-A)%B)+B)%B
(註: 以上關於取模的分析過程均采用大寫,關於不定方程的分析過程均采用小寫)
代入\(-A=x_0, B=\frac{b}{gcd(a,b)}\)即可,再將a,b分別換成原方程中的\(m-n\)和\(p\),直接暢通無阻地使用exgcd即可。
部分易錯點
- 很容易炸long long, 一定註意。
代碼實現
(三個OJ均AC)
#include<cstdio>
#include<algorithm>
using namespace std;
long long u,v,m,n,p;
long long exgcd(long long a,long long b,long long &x,long long &y)
{
if(b==0ll) {x = 1ll; y = 0ll; return a;}
long long ret = exgcd(b,a%b,y,x); y -= a/b*x;
return ret;
}
long long gcd(long long a,long long b)
{
if(b==0ll) return a;
else return gcd(b,a%b);
}
long long absl(long long x)
{
return x>0ll ? x : -x;
}
void swap_ll(long long &x,long long &y)
{
long long c = x; x = y; y = c;
}
int main()
{
scanf("%lld%lld%lld%lld%lld",&u,&v,&m,&n,&p);
long long x,y;
if(m==n) {puts("Impossible"); return 0;}
if(m-n<0) {swap_ll(u,v); swap_ll(m,n);}
if(absl(v-u)%gcd(m-n,p)!=0) {puts("Impossible"); return 0;}
exgcd(m-n,p,x,y);
long long s = x*((v-u)/gcd(m-n,p)); //此處一定是用(v-u)/gcd(m-n,p),x不一定被gcd整除
long long g = p/gcd(m-n,p); //把g直接當成了p使用,在BZOJ和POJ居然AC,所幸Luogu WA
s = ((s%g)+g)%g;
printf("%lld\n",s);
return 0;
}
POJ 1061 BZOJ 1477 Luogu P1516 青蛙的約會 (擴展歐幾裏得算法)