poj2115[扩展欧几里德]

C Looooops

Time Limit: 1000MS		Memory Limit: 65536K
Total Submissions: 22260		Accepted: 6125

Description

A Compiler Mystery: We are given a C-language style for loop of type

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

  statement;

I.e., a loop which starts by setting variable to value A and while variable is not equal to B, repeats statement followed by increasing the variable by C. We want to know how many times does the statement get executed for particular values of A, B and C, assuming that all arithmetics is calculated in a k-bit unsigned integer type (with values 0 <= x < 2^k) modulo 2^k.

Input

The input consists of several instances. Each instance is described by a single line with four integers A, B, C, k separated by a single space. The integer k (1 <= k <= 32) is the number of bits of the control variable of the loop and A, B, C (0 <= A, B, C < 2^k) are the parameters of the loop.

The input is finished by a line containing four zeros.

Output

The output consists of several lines corresponding to the instances on the input. The i-th line contains either the number of executions of the statement in the i-th instance (a single integer number) or the word FOREVER if the loop does not terminate.

Sample Input

3 3 2 16
3 7 2 16
7 3 2 16
3 4 2 16
0 0 0 0

Sample Output

0
2
32766
FOREVER

Source

CTU Open 2004

 

 

大致题意：

对于C的for(i=A ; i!=B ;i +=C)循环语句，问在k位存储系统中循环几次才会结束。

若在有限次内结束，则输出循环次数。

否则输出死循环。

解题思路：

题意不难理解，只是利用了 k位存储系统 的数据特性进行循环。

例如int型是16位的，那么int能保存2^16个数据，即最大数为65535（本题默认为无符号），

当循环使得i超过65535时，则i会返回0重新开始计数

如i=65534，当i+=3时，i=1

其实就是 i=(65534+3)%(2^16)=1

有了这些思想，设对于某组数据要循环x次结束，那么本题就很容易得到方程：

x=[(B-A+2^k)%2^k] /C

即 Cx=(B-A)(mod 2^k)  此方程为 模线性方程，本题就是求X的值。

下面将结合《算法导论》第2版进行简述，因此先把上面的方程变形，统一符号。

令a=C  

  b=B-A 

  n=2^k

那么原模线性方程变形为：

 ax=b (mod n)

该方程有解的充要条件为 gcd(a,n) | b ，即 b% gcd(a,n)==0

令d=gcd(a,n)

有该方程的 最小整数解为 x = e (mod n/d)

其中e = [x0 mod(n/d) + n/d] mod (n/d) ，x0为方程的最小解

那么原题就是要计算b% gcd(a,n)是否为0，若为0则计算最小整数解，否则输出FOREVER

当有解时，关键在于计算最大公约数 d=gcd(a,n) 与 最小解x0

参考《算法导论》，引入欧几里得扩展方程  d=ax+by ，

通过EXTENDED_EUCLID算法（P571）求得d、x、y值，其中返回的x就是最小解x0，求d的原理是辗转相除法（欧几里德算法）

再利用MODULAR-LINEAR-EQUATION-SOLVER算法（P564）通过x0计算x值。注意x0可能为负，因此要先 + n/d 再模n/d。

以上方法的推导过程大家自己看《算法导论》。。。这里不证明，只直接使用。

注意：

计算n=2^k时，用位运算是最快的，1<<k （1左移k位）就是2^k

但是使用long long的同学要注意格式， 1ll<<k

使用__int64的同学要强制类型转换 (__int64)1<<k

不然会WA

 TLE代码：

#include<cstdio>
#include<iostream>
using namespace std;
#define ll long long
ll gcd(ll a,ll b){
    if(!b) return a;
    return gcd(b,a%b);
}
ll quick_pow(ll x,ll n){
    if(n==0) return 1;
    else{
        while(!(n&1)){
            n>>=1;
            x*=x;
        }
    }
    ll result=x;
    n>>=1;
    while(n){
        x*=x;
        if((n&1)){
            result*=x;
        }
        n>>=1;
    }
    return result;
}
int main(){
    ll a,b,c,k;
    while(scanf("%I64d%I64d%I64d%I64d",&a,&b,&c,&k)==4){
        if(!a&&!b&&!c&&!k) break;
        ll d=gcd(c,b-a);
        if(b%d==0){puts("FOREVER");continue;}
        ll mod=quick_pow(2,k);
        printf("%I64d
",(b-a+mod)%mod/c);
    }
    return 0;
}

 AC代码：

#include<iostream>
#include<cstdio>
using namespace std;
#define LL long long
//d=ax+by,其中最大公约数d=gcd(a,n)，x、y为方程系数，返回值为d、x、y
LL gcd(LL a,LL b,LL& x,LL& y){
    if(b==0){
        x=1;y=0; //d=a，x=1,y=0,此时等式d=ax+by成立
        return a;
    }
    LL d=gcd(b,a%b,x,y);
    LL xt=x;
    x=y;
    y=xt-a/b*y;//系数x、y的取值是为满足等式d=ax+by
    return d;
}
int main(){
    LL A,B,C,k;
    while(scanf("%I64d%I64d%I64d%I64d",&A,&B,&C,&k)==4){
        if(!A&&!B&&!C&&!k) break;
        LL a=C;
        LL b=B-A;
        LL n=1ll<<k;
        LL x,y;
        LL d=gcd(a,n,x,y);//求a,n的最大公约数d=gcd(a,n)和方程d=ax+by的系数x、y
        if(b%d!=0) puts("FOREVER");//方程 ax=b(mod n) 无解
        else{
            x=(x*(b/d))%n;//方程ax=b(mod n)的最小解
            x=(x%(n/d)+n/d)%(n/d);//方程ax=b(mod n)的最整数小解
            printf("%I64d
",x);
        }
    }
    return 0;
}