C Looooops POJ - 2115
扩展欧几里得算法
题目描述:算循环的次数,初始值为A,跳出循环的条件是A!=B,A每次+C,设所有的运算都是二进制k位数,运用补码的原理,当x=2^k,则x+1溢出后=1.
解题思路:扩展欧几里得算法。推一下方程,令所求的次数为x (A+Cx)%2^k=B ->
A+Cx=2^k*y+B ->移项得Cx-2^k*y=B-A; 将各参数代入函数即可。
代码如下:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
long long exgcd(long long a,long long b,long long &x,long long &y)
{
if(b==0)
{
x=1;
y=0;
return a;
}
long long ans=exgcd(b,a%b,x,y);
long long t=y;
y=x-(a/b)*y;
x=t;
return ans;
}
int main()
{
int A,B,C,K;
while(scanf("%d %d %d %d",&A,&B,&C,&K)==4&&(A||B||C||K))
{
long long a=C,b=(long long)1<<K,c=B-A,x,y;
long long gcd=exgcd(a,b,x,y);
if(c%gcd)
{
printf("FOREVER\n");
}
else
{
long long x0=x*(c/gcd)%(b/gcd);
x=(x0+(b/gcd))%(b/gcd);
printf("%I64d\n",x);
}
}
return 0;
}