POJ-2115 C Looooops（扩展欧几里得）

题意

c++语言的for循环格式如下：

for (variable = A; variable != B; variable += C)

  statement;

这里的 $variable$ 是一个 $k$ 位的无符号整数，也就是说，当位数大于k时，高位会溢出并舍去，求这个循环进行了多少次。

思路

设这个循环进行了 $k_1$ 次，不难列出方程 $A+Ck_1=B(mod$ $2^k)$ ，化简后得到 $Ck_1+2^kk_2=B-A$ ， 进行Exgcd解出 $k_1$ 即可。

代码

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
typedef long long LL;
using namespace std;

LL gcd(LL a,LL b){return b?gcd(b,a%b):a;}
void exgcd(LL a,LL b,LL &x,LL &y)
{
    if(!b){x=1,y=0;return;}
    exgcd(b,a%b,y,x);y-=a/b*x;
    return;
}
LL Exgcd(LL A,LL B,LL C)
{
    LL k1,k2,g=gcd(A,B);
    if(C%g)return -1;
    A/=g,B/=g,C/=g;
    exgcd(A,B,k1,k2);
    return (k1*C%B+B)%B;
}

int main()
{
    LL A,B,C,k;
    while(scanf("%lld%lld%lld%lld",&A,&B,&C,&k),(A|B|C|k))
    {
        LL k1=Exgcd(C,1LL<<k,B-A);
        ~k1?printf("%lld\n",k1):printf("FOREVER\n");
    }
    return 0;
}