扩展欧几里得
小Hi和小Ho周末在公园溜达。公园有一堆围成环形的石板,小Hi和小Ho分别站在不同的石板上。已知石板总共有m块,编号为 0..m-1,小Hi一开始站在s1号石板上,小Ho一开始站在s2号石板上。
小Hi:小Ho,你说我们俩如果从现在开始按照固定的间隔数同时同向移动,我们会不会在某个时间点站在同一块石板上呢?
小Ho:我觉得可能吧,你每次移动v1块,我移动v2块,我们看能不能遇上好了。
小Hi:好啊,那我们试试呗。
一个小时过去了,然而小Hi和小Ho还是没有一次站在同一块石板上。
小Ho:不行了,这样走下去不知道什么时候才汇合。小Hi,你有什么办法算算具体要多久才能汇合么?
小Hi:让我想想啊。。
Iput
第1行:每行5个整数s1,s2,v1,v2,m,0≤v1,v2≤m≤1,000,000,000。0≤s1,s2<m
中间过程可能很大,最好使用64位整型
Output
第1行:每行1个整数,表示解,若该组数据无解则输出-1
分析:
提示:扩展欧几里德
小Hi:首先可以我俩现在的情况列出一个式子:
s1+v1*t=s2+v2*t-k*m (v1<v2)
也就是经过t时间过后,速度快的人刚好超过了速度慢的人k圈,且到达同一个位置。
将这个式子进行变换得到:
(v1-v2)*t+k*m=(s2-s1)
即原式子变成了形如”Ax+By=C”的情况,我们要求解的是一组(x,y)使得原公式成立。
小Ho:形如”Ax+By=C”,也就是说我们令A=(v1-v2),B=m,C=(s2-s1),x=t,y=k。
小Hi:恩,没错。
求解该式子的算法我们称为扩展欧几里德算法。
该算法分为两个部分:
(1) 判定是否存在解
对于形如”Ax+By=C”的式子,其存在解的条件为C为A和B最大公约数的整数倍。
我们将A和B的最大公约数记为gcd(A,B)。因此其有解的条件是C=n*gcd(A,B)。
那么我们应该如何来求解gcd(A,B)呢?
一个朴素的算法是枚举1~min(A,B),最大的一个能同时被A,B整除的数即gcd(A,B)。显然这个算法是非常没有效率的。
为了求解gcd(A,B),欧几里德提出了一个辗转相除法:
首先要证明一个定理:gcd(A,B) = gcd(B, A mod B)
证明: 假设A = k*B+r,有r = A mod B。不妨设d为A和B的一个任意一个公约数,则有A = pd, B = qd。 由r = A - k*B = pd - k*qd = (p - kq)*d,所以有d也为r的约数,因此d是B和A mod B的公约数。 由于对任意一个A和B的公约数都满足这个性质,gcd(A,B)也满足,因此有gcd(A,B)=gcd(B,A mod B)。
利用这个性质,我们可以得到算法:
A mod B = 0, 则B为gcd(A,B) A mod B ≠ 0, 则gcd(A,B) = gcd(B, A mod B)
通过不断的模运算,数据的规模也越来越小,因此能够快速得收敛到一个解。将其写成伪代码为:
gcd(A, B): If (A mod B == 0) Then Return B End If Return gcd(B, A mod B)
(2) 求解
在判定有解之后,我们需要在其基础上再求出一组(x,y)。由于A,B,C均是gcd(A,B)的整数倍,因此可以将它们都缩小gcd(A,B)倍。即A’=A/gcd(A,B),B’=B/gcd(A,B),C’=C/gcd(A,B)。
化简为A’x+B’y=C’,gcd(A’,B’)=1,即A’,B’互质。
此时,我们可以先求解出A’x+B’y=1的解(x’,y’),再将其扩大C’倍,即为我们要求的最后解(x,y)=(C’x’, C’y’)。
那么接下来我们来研究如何求解A’x+B’y=1:
假设A>B>0,同时我们设:
A * x[1] + B * y[1] = gcd(A, B) B * x[2] + (A mod B) * y[2] = gcd(B, A mod B)
已知gcd(A,B)=gcd(B, A mod B),因此有:
A * x[1] + B * y[1] = B * x[2] + (A mod B) * y[2] => A * x[1] + B * y[1] = B * x[2] + (A - kB) * y[2] // A = kB + r => A * x[1] + B * y[1] = A * y[2] + B * x[2] - kB * y[2] => A * x[1] + B * y[1] = A * y[2] + B * (x[2] - ky[2]) => x[1] = y[2], y[1] = (x[2] - ky[2])
利用这个性质,我们可以递归的去求解(x,y)。
其终止条件为gcd(A, B)=B,此时对应的(x,y)=(0,1)
将这个过程写成伪代码为:
extend_gcd(A, B): If (A mod B == 0) Then Return (0, 1) End If (tempX, tempY) = extend_gcd(B, A mod B) x = tempY y = tempX - (A / B) * tempY Return (x, y)
小Ho:那么我只需要把A=(v1-v2),B=m,C=(s2-s1),x=t,y=k代入就可以得到t了么?
小Hi:是的,在已知A,B,C的情况下,我们的确能够顺利求解出一组合法的(x,y)。
但是在求解过程中,我们并没有保证x是最小的非负整数,它不能直接作为我们的解。
小Ho:那还需要做怎样的处理么?
小Hi:我们需要将(A’,B’,x’,y’)扩充为一个解系。
由于A’B’是互质的,所以可以将A’x’+B’y’=1扩展为:
A'x'+B'y'+(u+(-u))A'B'=1 => (x' + uB')*A' + (y' - uA')*B' = 1 => X = x' + uB', Y = y' - uA'
可以求得最小的X为(x’+uB’) mod B’,(x’+uB’>0)
同时我们还需要将X扩大C’倍,因此最后解为:
x = (x'*C') mod B'
若x<0,则不断累加B’,直到x>0为止。
那么最后,小Ho你来总结一下主体部分的伪代码吧!
小Ho:好的,最后的代码为:
solve(s1, s2, v1, v2, m): A = v1 - v2 B = m C = s2 - s1 If (A < 0) Then A = A + m // 相对距离变化 End If D = gcd(A, B) If (C mod D) Then Return -1 End If A = A / D B = B / D C = C / D (x, y) = extend_gcd(A, B) x = (x * C) mod B While (x < 0) x = x + B End While Return x
代码:
#include "iostream"
#include "cstdio"
using namespace std;
long long gcd(long long A,long long B) //最大公约数
{
while(A=A%B)
{
long long C;
C=A;
A=B;
B=C;
}
return B;
}
void extend_gcd(long long A,long long B,long long &x,long long &y)
{
if(A%B == 0){
x=0;
y=1;
return;
}
long long tempX,tempY;
extend_gcd(B, A%B,tempX,tempY);
x = tempY;
y = tempX - (A / B) * tempY;
return;
}
int solve(long long s1,long long s2,long long v1,long long v2,long long m)
{
long long A = v1 - v2;
long long B = m;
long long C = s2 - s1;
if(A < 0)
A = A + m;
long long D = gcd(A, B);
if(C%D)
return -1;
A = A / D;
B = B / D;
C = C / D;
long long x,y;
extend_gcd(A,B,x,y);
x = (x * C)%B;
while(x < 0)
{
x = x + B;
}
return x;
}
int main()
{
long long s1,s2,v1,v2;
long long m;
while(scanf("%lld%lld%lld%lld%lld",&s1,&s2,&v1,&v2,&m)!=EOF)
cout<<solve(s1,s2,v1,v2,m)<<endl;
return 0;
}