SGU 141 Jumping Joe(扩展欧几里得)

题目链接:http://acm.sgu.ru/problem.php?contest=0&problem=141

题意:给出x1,x2,P,K,求一组非负整数解(p1,n1,p2,n2)满足:p1+n1+p2+n2=K,(p1-n1)*x1+(p2-n2)*x2=P。

思路:令x=p1-n1,y=p2-n2,可求得x1*x+x2*y=P的一组解(x,y),令temp=Gcd(x1,x2),x=x*P/temp,y=y*P/temp,d1=x2/temp,d2=x1/temp,则通解为X=x+d1*k,Y=y-d2*k。进而可以算出使得abs(X)+abs(Y)最小的X和Y。若abs(X)+abs(Y)>K则无解,因为此时的X和Y是最小需要的步数。接着,若K-abs(X)-abs(Y)为偶数,则将这些剩下的步数分为两半,来回各走一半就可解决。若K-abs(X)-abs(Y)为奇数,那么我们需要使用d1和d2使得K-abs(X)-abs(Y)为偶数。此时d1+d2为偶数时无解;否则,X和Y再加减或减加一次d1、d2就可以解决。





int x1,x2,P,K;





int exGcd(int a,int b,int &x,int &y)

{

    if(!b)

    {

        x=1;y=0;

        return a;

    }

    int r=exGcd(b,a%b,x,y);

    int temp=x;

    x=y;

    y=temp-a/b*y;

    return r;

}



int main()

{

    RD(x1,x2);

    RD(P,K);

    int x,y,temp=exGcd(x1,x2,x,y);

    if(abs(P)%abs(temp))

    {

        puts("NO");

        return 0;

    }

    x*=P/temp;

    y*=P/temp;

    int d1=x2/temp,d2=x1/temp;

    while(abs(x+d1)+abs(y-d2)<abs(x)+abs(y))

    {

        x+=d1;

        y-=d2;

    }

    while(abs(x-d1)+abs(y+d2)<abs(x)+abs(y))

    {

        x-=d1;

        y+=d2;

    }

    if(abs(x)+abs(y)>K)

    {

        puts("NO");

        return 0;

    }

    if((K-abs(x)-abs(y))&1)

    {

        if(abs(d1+d2)%2==0)

        {

            puts("NO");

            return 0;

        }

        if(abs(x+d1)+abs(y-d2)<abs(x-d1)+abs(y+d2))

        {

            x+=d1;

            y-=d2;

        }

        else

        {

            x-=d1;

            y+=d2;

        }

        if(abs(x)+abs(y)>K)

        {

            puts("NO");

            return 0;

        }

    }

    int t=(K-abs(x)-abs(y))>>1;

    int p1,p2,n1,n2;

    if(x<0) p1=t,n1=t+abs(x);

    else p1=t+x,n1=t;

    if(y<0) p2=0,n2=-y;

    else p2=y,n2=0;

    puts("YES");

    printf("%d %d %d %d\n",p1,n1,p2,n2);

    return 0;

}

  

你可能感兴趣的:(ping)