【BZOJ2144】Throw 数论

题目大意

　　给你三个数$a,b,c$，每次你可以选择一个数$s_1$，再选择一个数$s_2$，把$s_1$变成$2s_2-s_1$，但要求$s_3$不在$s_1$到$2s_2-s_1$之间。

　　再给你三个数$x,y,z$，问你是否能把$a,b,c$变成$x,y,z$。

　　$|a|,|b|,|c|,|x|,|y|,|z|\leq {10}^9$

题解

　　首先三个数$a,b,c(a<b<c)$只能有三种转移：

$$(a,b,c)\rightarrow \begin{cases} (2a-b,a,c)\\ (a,c,2c-b)\\ (b,2b-a,c)~~~~~(c-b>b-a)\\ (a,2b-c,b)~~~~~(c-b<b-a) \end{cases}$$
　　可以发现，这些状态构成了一棵树。

　　转移可以用辗转相除法加速。

　　直接暴力往上跳找LCA即可。

　　时间复杂度：$O(\log (\max_a-\min_a))$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<ctime>
#include<utility>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
void sort(ll &a,ll &b,ll &c)
{
    if(b<a)
        swap(a,b);
    if(c<a)
        swap(a,c);
    if(c<b)
        swap(b,c);
}
void open(const char *s)
{
#ifdef DEBUG
    char str[100];
    sprintf(str,"%s.in",s);
    freopen(str,"r",stdin);
    sprintf(str,"%s.out",s);
    freopen(str,"w",stdout);
#endif
}
ll gcd(ll a,ll b)
{
    return b?gcd(b,a%b):a;
}
ll getmid(ll a,ll b,ll c)
{
    while(b-a!=c-b)
        if(b-a>c-b)
        {
            ll j=(b-a)/(c-b);
            ll k=c-b;
            c-=j*k;
            b-=j*k;
            if(b==a)
            {
                b+=k;
                c+=k;
            }
        }
        else
        {
            ll j=(c-b)/(b-a);
            ll k=b-a;
            a+=j*k;
            b+=j*k;
            if(b==c)
            {
                b-=k;
                a-=k;
            }
        }
    return b;
}
pll a1[110];
int d1[110];
pll a2[110];
int d2[110];
int cnt1,cnt2;
int main()
{
    open("c");
    ll a,b,c,x,y,z;
    scanf("%lld%lld%lld%lld%lld%lld",&a,&b,&c,&x,&y,&z);
    sort(a,b,c);
    sort(x,y,z);
    if(gcd(b-a,c-b)!=gcd(y-x,z-y))
    {
        printf("NO\n");
        return 0;
    }
    if(getmid(a,b,c)!=getmid(x,y,z))
    {
        printf("NO\n");
        return 0;
    }
    printf("YES\n");
    pll s1(b-a,c-b);
    pll s2(y-x,z-y);
    cnt1=cnt2=0;
    d1[0]=d2[0]=0;
    while(s1.first!=s1.second)
    {
        a1[++cnt1]=s1;
        if(s1.first>s1.second)
        {
            d1[cnt1]=s1.first/s1.second;
            s1.first%=s1.second;
            if(!s1.first)
            {
                s1.first=s1.second;
                d1[cnt1]--;
            }
        }
        else
        {
            d1[cnt1]=s1.second/s1.first;
            s1.second%=s1.first;
            if(!s1.second)
            {
                s1.second=s1.first;
                d1[cnt1]--;
            }
        }
    }
    a1[++cnt1]=s1;
    d1[cnt1]=0;
    reverse(a1+1,a1+cnt1+1);
    reverse(d1+1,d1+cnt1+1);
    while(s2.first!=s2.second)
    {
        a2[++cnt2]=s2;
        if(s2.first>s2.second)
        {
            d2[cnt2]=s2.first/s2.second;
            s2.first%=s2.second;
            if(!s2.first)
            {
                s2.first=s2.second;
                d2[cnt2]--;
            }
        }
        else
        {
            d2[cnt2]=s2.second/s2.first;
            s2.second%=s2.first;
            if(!s2.second)
            {
                s2.second=s2.first;
                d2[cnt2]--;
            }
        }
    }
    a2[++cnt2]=s2;
    d2[cnt2]=0;
    reverse(a2+1,a2+cnt2+1);
    reverse(d2+1,d2+cnt2+1);
    ll ans=0;
    int i,j;
    for(i=1;i<=cnt1;i++)
        d1[i]+=d1[i-1];
    for(i=1;i<=cnt2;i++)
        d2[i]+=d2[i-1];
    i=cnt1,j=cnt2;
    while(a1[i-1]!=a2[j-1])
        if(d1[i-1]>d2[j-1])
        {
            ans+=d1[i]-d1[i-1];
            i--;
        }
        else
        {
            ans+=d2[j]-d2[j-1];
            j--;
        }
    if(a1[i].first==a2[j].first||a1[i].second==a2[j].second)
    {
        if(d1[i]>d2[j])
            ans+=d1[i]-d2[j];
        else
            ans+=d2[j]-d1[i];
    }
    else
        ans+=d1[i]-d1[i-1]+d2[j]-d2[j-1];
    printf("%lld\n",ans);
    return 0;
}