数论-裴蜀定理-POJ2115-C-Loop

数论-裴蜀定理-POJ2115-C-Loop

题目：

对于 C 语言的循环语句，形如：

for (variable = A; variable != B; variable += C)
statement;
请问在 k 位存储系统中循环几次才会结束。

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

输入格式
多组数据，每组数据一行四个整数 A,B,C,k。

读入以 0 0 0 0 结束。

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

否则输出 FOREVER。

数据范围
1≤k≤32,
0≤A,B,C<2k
输入样例：
3 3 2 16
3 7 2 16
7 3 2 16
3 4 2 16
0 0 0 0
输出样例：
0
2
32766
FOREVER

题意：

$k位系统中二进制的值仅保留后k位，即需要将值对2^k取模。$

$列出关系式：(A+Cx)mod{2}^k=B等价于A+Cx+2^{k}y=B，即求方程Cx+2^{k}y=B-A的解(x,y)。$

$①、判断ax+by=d是否有整数解即判断gcd(a,b)是否为d的因子。$

$②、要求x的最小正值，通过ax+by=d的通解形式\{\begin{array}{l}x=x_0+k{b}^{\prime }\\ y=y_0-k{a}^{\prime }\end{array},其中a^{\prime }=\frac{a}{d},b^{\prime }=\frac{b}{d}。$
$将x对b^{\prime }取模，最小正值即(x_0\%b^{\prime }+b^{\prime })\%b^{\prime }。$

解题步骤：

1. $扩展欧几里得算法求方程Cx+2^{k}y=gcd(C,2^k)的解，若gcd(C,2^k)不是B-A的因子，则死循环。$
2. $接着将求得的x扩大\frac{B-A}{gcd(C,2^k)}倍。$
3. $最后根据公式求最小值(x\%b^{\prime }+b^{\prime })\%b^{\prime }。$

---

代码：

#include<cstdio>
#include<iostream>
#define ll long long

using namespace std;

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

int main()
{
    ll A,B,C,k;
    while(~scanf("%lld%lld%lld%lld",&A,&B,&C,&k),A||B||C||k)  
    {
        ll x,y;
        ll b=1ll<<k;   ///1要转换为ll
        int d=exgcd(C,b,x,y);
        if((B-A)%d!=0) cout<<"FOREVER"<<endl;
        else
        {
            int tmp=((B-A)/d);
            x*=tmp;
            b/=d;  ///b要更新为b'
            x=(x%b+b)%b;
            cout<<x<<endl;
        }
    }

    return 0 ;
}