C Looooops poj 2142 扩展欧几里德算法

题目大意

对于for(i=A ; i!=B ;i+=C)循环语句，问在k位存储系统中循环几次才会结束。
比如：当k=4时，存储的数 i 在0-15之间循环。(本题默认为无符号)
若在有限次内结束，则输出循环次数。
否则输出死循环。

分析

本题利用扩展欧几里德算法求线性同余方程，设循环次数为 x ,则解方程 (A + C*x) % 2^k = B ;求出最小正整数 x。
　　1，化简方程化为求线性同余方程标准式 ax ≡ b (mod n);
　　2，扩展欧几里德算法求解线性同余方程 C*x ≡ B-A (mod 2^k);
　　3，求出最小非负整数解。
步骤：
　　1，化简：(A + C*x) mod 2^K = B –> C*x mod 2^k = B-A –> C*x ≡ B-A (mod 2^k);
　　2，求线性同余方程 C* x ≡ B-A (mod 2^k) , 就相当于求二元一次方程 C*x + 2^k * y = B-A
　　i,代入扩展欧几里德算法，求解方程 C*x + 2^k * y = gcd(C , 2^k) ;
　　ii,利用方程 C* x + 2^k * y = gcd(C , 2^k)的解 x0 以及公式 x1 = x0 * c/d 求出原方程 a*x + b*y = c 的解 x1 ;前提是：d|c (c 能被 d 整除);

code

#include<iostream>
#include<cstring>
#include<cstdio>

using namespace std;

__int64 exgcd(__int64 a,__int64 b,__int64& x,__int64& y)
{
    if (b==0)
    {
        x=1;
        y=0;
        return a;
    }
    int d=exgcd(b,a%b,y,x);
    y=y-(a/b)*x;
    return d;
}

int main()
{
    __int64 a,b,c,k,d,x,y,t;
    a=0; b=0; c=0; k=0; d=0; x=0; y=0; t=0;
    cin>>a>>b>>c>>k;
    while ((a!=0)||(b!=0)||(c!=0)||(k!=0))
    {
        __int64 n=(__int64)1<<k;
        x=0; y=0;
        d=exgcd(c,n,x,y);
        if ((b-a)%d!=0)
            printf("FOREVER\n");
        else
        {
            x=x*(b-a)/d;
            t=n/d;
            x=(x%t+t)%t;
            cout<<x<<endl;
        }
        cin>>a>>b>>c>>k;
    }
}