BZOJ2144跳跳棋——LCA+二分

题目描述

跳跳棋是在一条数轴上进行的。棋子只能摆在整点上。每个点不能摆超过一个棋子。我们用跳跳棋来做一个简单的
游戏:棋盘上有3颗棋子,分别在a,b,c这三个位置。我们要通过最少的跳动把他们的位置移动成x,y,z。(棋
子是没有区别的)跳动的规则很简单,任意选一颗棋子,对一颗中轴棋子跳动。跳动后两颗棋子距离不变。一次只
允许跳过1颗棋子。

写一个程序,首先判断是否可以完成任务。如果可以,输出最少需要的跳动次数。

输入

第一行包含三个整数,表示当前棋子的位置a b c。(互不相同)
第二行包含三个整数,表示目标位置x y z。(互不相同)

输出

如果无解,输出一行NO。如果可以到达,第一行输出YES,第二行输出最少步数。

样例输入

1 2 3
0 3 5

样例输出

YES
2
【范围】
100% 绝对值不超过10^9
 
思维神题。
考虑对于当前状态的a,b,c有哪些可移动方案,设d1=b-a,d2=c-b,如果d1!=d2,那么b可以向两边跳,d1,d2其中小的那个可以向中间跳;如果d1=d2那么只能由b向两边跳。
可移动方案最多只有三种,那么可以将每个状态看成一个点,往左右跳看作这个点的左右子节点,往中间跳看作是这个点的父节点,如果不能往中间跳,那这个点就是根节点。
那么所有状态就变成了一个二叉树森林,判断能否完成就变成了判断两个状态是否在同一棵树中,而最小步数自然就是两点间的距离了。
但如果将所有状态都枚举出来显然不行,例如下面这个样例:
1 2 1e9
1e9-1 1e9-2 1e9
要跳1e9级别这么多次,显然不能暴力跳。
那么再回到求答案的那一步,两点间的距离不就是lca分别和两点深度差的和吗!
而深度就是每个点跳到根节点的步数。
那么两点往上跳在原题中就是两边的点往中间跳。
因为跳的点和被跳的点之间的相对距离不变,那么就相当于将两个点都平移了两点间距离这么多。
假设d1>d2,那么c最多向左平移(d1-1)/d2次(因为不能跳到同一个点)。
对于d1和d2,我们可以像求gcd一样辗转相除来求得在二叉树上给出的这两点的深度,然后将深度深的点往上跳使两点深度相同。
接下来只要找到深度相同的这两个点的lca就好了,可以像求倍增lca一样往上跳验证,也可以用二分答案来往上跳验证。
我这里用了二分的写法。注意原题三个数不一定按顺序给出。
#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<vector>
#include<bitset>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
ll a,b,c;
ll x,y,z;
ll dep1,dep2;
ll root1,root2;
ll l1,l2;
ll len;
void cmp(ll &a,ll &b,ll &c)
{
    if(a>b)
    {
        swap(a,b);
    }
    if(a>c)
    {
        swap(a,c);
    }
    if(b>c)
    {
        swap(b,c);
    }
}
ll find_root(ll a,ll b,ll c,ll &dep,ll &anc)
{
    ll d1=b-a;
    ll d2=c-b;
    while(d1!=d2)
    {
        if(d1<d2)
        {
            ll s=d2/d1;
            ll t=d2%d1;
            if(t==0)
            {
                dep+=(s-1);
                anc=d1;
                return a+(s-1)*d1;
            }
            else
            {
                dep+=s;
                a+=s*d1;
                d2=t;
            }
        }
        else
        {
            ll s=d1/d2;
            ll t=d1%d2;
            if(t==0)
            {
                dep+=(s-1);
                anc=d2;
                return a;
            }
            else
            {
                dep+=s;
                d1=t;
            }
        }
    }
    dep=0;
    anc=d1;
    return a;
}
void get_fa(ll &a,ll &b,ll &c,ll dep)
{
    ll d1=b-a;
    ll d2=c-b;
    while(dep>0)
    {
        if(d1<d2)
        {
            ll s=d2/d1;
            ll t=d2%d1;
            if(s>=dep)
            {
                a+=dep*d1;
                b+=dep*d1;
                if(b==c)
                {
                    b=a;
                    a-=d1;
                }
                return ;
            }
            else
            {
                dep-=s;
                a+=s*d1;
                b+=s*d1;
                d2=t;
            }
        }
        else
        {
            ll s=d1/d2;
            ll t=d1%d2;
            if(s>=dep)
            {
                c-=dep*d2;
                b-=dep*d2;
                if(a==b)
                {
                    b=c;
                    c+=d2;
                }
                return ;
            }
            else
            {
                dep-=s;
                b-=s*d2;
                c-=s*d2;
                d1=t;
            }
        }
    }
}
int main()
{
    scanf("%lld%lld%lld",&a,&b,&c);
    scanf("%lld%lld%lld",&x,&y,&z);
    cmp(a,b,c);
    cmp(x,y,z);
    l1=find_root(a,b,c,dep1,root1);
    l2=find_root(x,y,z,dep2,root2);
    if(l1!=l2||root1!=root2)
    {
        printf("NO");
        return 0;
    }
    if(dep1<dep2)
    {
        len+=dep2-dep1;
        get_fa(x,y,z,len);
    }
    else
    {
        len+=dep1-dep2;
        get_fa(a,b,c,len);
    }
    ll l=0;
    ll r=min(dep1,dep2);
    ll ans=0;
    while(l<=r)
    {
        ll mid=(l+r)/2;
        ll a1=a,b1=b,c1=c;
        ll x1=x,y1=y,z1=z;
        get_fa(a1,b1,c1,mid);
        get_fa(x1,y1,z1,mid);
        if(a1==x1&&b1==y1&&c1==z1)
        {
            ans=mid;
            r=mid-1;
        }
        else
        {
            l=mid+1;
        }
    }
    printf("YES
");
    printf("%lld",len+ans*2);
}
原文地址:https://www.cnblogs.com/Khada-Jhin/p/9776977.html