POJ 2115 C Looooops （扩展欧几里得+同余运算性质）

题目链接：POJ 2115 C Looooops

题意比较好懂。就是给出一个for循环求算执行多少步，每一步都要取模。

转化成扩展欧几里得求逆元的问题：A+C*X≡ B mod 2^k  等价于  C*X≡ （B-A） mod 2^k 。（同余的线性运算）。接下来就是就C关于（B-A）模（2^k）的逆元。

注意几种等于0的情况。

AC代码：

#include<stdio.h>
#define ll __int64

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

//a*x==c mod b;
//a+c*x== b mod 2^k, a==a mod 2^k, c*x== b-a mod 2^k 
ll cal(ll a,ll b,ll c)
{
	ll x,y;
	ll gcd=exgcd(a,b,x,y);
	if(c%gcd) return -1;
	x*=c/gcd;
	if(b<0)
		b=-b;
	ll ans=x%b;
	if(ans<=0) ans+=b;
	ans=ans%(b/gcd);
	return ans;//ans%(b/gcd) 最小整数解
}
int main()
{
	ll a,b,c,k;
	while(scanf("%I64d %I64d %I64d %I64d",&a,&b,&c,&k)!=EOF)
	{
		if(a==0 && b==0 && c==0 && k==0)
			break;
		if(a==b)
		{
			printf("0\n");
			continue;
		}
		ll kmod=ll(1)<<k;
		if(c==0)
		{
			if(a%kmod==b)
				printf("1\n");
			else
				printf("FOREVER\n");
			continue;
		}
		ll temp=(b-a);
		while(temp<0)
			temp+=kmod;
		ll ans=cal(c,kmod,temp);
		if(ans==-1)
			printf("FOREVER\n");
		else
			printf("%I64d\n",ans);
	}
return 0;
}
/*
0 2 2 16
3 1 0 1
0 0 0 1
2 0 1 1
3 0 1 1
2 2 0 1

*/