poj 2115——Looooops

题目大意：循环变量i从A开始，每次加上C，当等于B的时候终止。求循环的次数，题目中给出数是用k位的无符号整数表示，即循环到1<<k的时候就自动回到0.如果死循环就输出FOREVER

思路：首先可列出方程 (A+nC)%2^k=B%2^k，然后变一下形式就变成了nC%2^k=(B-A)%2^k，就变成了单元模线性方程。所求答案就是n,该方程的最小解。题目中可能B<A，这个时候，因为B-A是负数，所以若有解，必然是变量i到达了一次2^k然后从0开始然后又到达的B。所以这个时候求的是方程的第二个解。

错误：第一次写的时候用的32位整数错了，后来在移位运算的时候因为k是64位，所以应该用1LL<<k。

代码如下：

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;

ll gcd(ll a,ll b,ll &x,ll &y){
    if(b==0){
        x=1;y=0;
        return a;
    }
    else{
        ll r=gcd(b,a%b,y,x);
        y-=x*(a/b);
        return r;
    }
}

ll line_mod_equation(ll a,ll b,ll n){
    ll x,y;
    ll d=gcd(a,n,x,y);
    if(b%d)return -1;
    x%=n;x+=n;x%=n;
    ll ans=x*(b/d)%(n/d);
    if(b>=0)
    return ans;
    else return (ans+n/d)%n;
}

int main(){
 //   freopen("data.txt","r",stdin);
    ll A,B,C,k;
    while(scanf("%lld%lld%lld%lld",&A,&B,&C,&k)==4){
        if(A==0&&B==0&&C==0&&k==0){break;}
        ll ans=line_mod_equation(C,B-A,1LL<<k);
        if(ans==-1)printf("FOREVER\n");
        else{
            printf("%lld\n",ans);
        }
    }
    return 0;
}