训练c 2

/*题目内容：
计算循环语句的执行频次 for (i = A; i != B; i += C) x += 1;
其中A, B, C, i都是k位无符号整数。
输入描述
A B C k, 其中0<k<32

输出描述
输出执行频次数，如果是无穷，则输出“forever”

输入样例
3 7 2 16

输出样例
2

求 ax+by=gcd(a,b)
引入 x1,y1
通过 exgcd(a,b,x,y)
得
x=y1;
y=x1-(a/b)*y1;
*/
#include
using namespace std;
typedef long long LL;

LL exgcd(LL a,LL b,LL &x,LL &y)
{
    if(b==0){
        x=1;
        y=0;
        return a;
    }
    LL x1,y1,ans;
    ans= exgcd(b,a%b,x1,y1);
    x=y1;
    y=x1-(a/b)*y1;
    return ans;
}

int main()
{
	LL A,B,C,k;
	LL a,b,c,x,y;
	cin>>A>>B>>C>>k; 
	a = C;
	b = 1LL<<k; // pow(2,k) 
	c = B - A;
	if(!A&&!B&&!C&&!k)  cout<<0<<endl;
	else {
		LL g =exgcd(a,b,x,y);
		if(c%g) cout<<"FOREVER"<<endl;
		else{
            b/=g;c/=g;
            cout<<(x%b*c%b+b)%b<<endl;
		}
	}
	return 0;
}