poj 2115 C Looooops (扩展欧几里德)

来源:互联网 发布:淘宝u站今日特惠 编辑:程序博客网 时间:2024/06/06 07:27

题意:对于C的for(i=A ; i!=B ;i +=C)循环语句,A,B,C为无符号整型,最大为2^k-1,问这个循环能否结束,如果能结束,输出循环次数,否则输出FOREVER
思路:设循环x次,(A+Cx)%2^k = B,则A+Cx=B+(2^k)y,y为未知数,移项后为Cx-(2^k)*y=B-A==>Cx+(2^k)(-y) = B-A,到这里就看出来用扩展欧几里德算法求解了。d=gcd(C,2^k),如果d能整除B-A则有解,结果就是最小正整数解,如果不能整除就无解。
计算过程中如果用移位运算符的话要指明1LL左移k位,如果是1左移k位的话,1会按照32位来算,当1<<31或1<<32的时候就会得到错误结果。我在这里wa了好几发,看了discuss才知道。。
下边是最小正整数解的证明过程:
  证明过程是朋友从网上找的。。。。。。
  关于 x 的模方程 ax%b=c 的解
  方程转换为 ax+by=c 其中 y 一般为非正整数
  则问题变为用 exgcd 解不定方程
  (gcd(a,b)=r,我自己加的)
  解得 x1=x0*c/r
  通解为 x=x1+b/r*t
  设 s=b/r (已证明 b/r 为通解的最小间隔)
  则 x 的最小正整数解为 (x1%s+s)%s
  证明:
    若 x1>0,则 (x1%s+s)%s=x1%s%s+s%s=x1%s=x1-ts (t∈N)
    若 x1<0,因在 C++ 里 a%b=-(-a%b)<0 (a<0 , b>0) 如 -10%4=-2
         则 (x1%s+s)%s=(-(-x1%s)+s)%s=(-(ts-x1)+s)%s=ts-x1 (t∈N)
    即为 x1 通过加或减上若干个 s 后得到的最小正整数解
    得证
  亦可伪证 x1<0 的情况:设 x1=-5 , s=2
             则 (x1%s+s)%s=(-5%2+2)%2=(-1+2)%2=3%2=1
             即为 x1 加上 3 个 s 后的到的最小正整数解

#include <cstdio>long long extend_gcd(long long a,long long b,long long &x,long long &y){    if(a==0&&b==0) return -1;    if(b==0)    {        x=1;        y=0;        return a;    }    long long d=extend_gcd(b,a%b,y,x);    y-=a/b*x;    return d;}int main(){    long long A,B,C,k,maxn,d,x,y,r,der,res;    while(scanf("%lld %lld %lld %lld",&A,&B,&C,&k) && (A+B+C+k))    {        if(C==0 && A!=B)        {            printf("FOREVER\n");            continue;        }        else if(A == B)        {            printf("0\n");            continue;        }        r = B-A;        maxn = 1LL<<k;        d = extend_gcd(C,maxn,x,y);        der = maxn/d;        res = (x*(r/d))%maxn;        if(r%d != 0)            printf("FOREVER\n");        else            printf("%lld\n",(res%der+der)%der);    }    return 0;}
0 0
原创粉丝点击