【JZOJ1917】【2011集训队出题】跳跳棋（树+二分）

Problem

　　跳跳棋是在一条数轴上进行的。棋子只能摆在整点上。每个点不能摆超过一个棋子。
　　我们用跳跳棋来做一个简单的游戏：棋盘上有3颗棋子，分别在a，b，c这三个位置。我们要通过最少的跳动把他们的位置移动成x，y，z。（棋子是没有区别的）
　　跳动的规则很简单，任意选一颗棋子，对一颗中轴棋子跳动。跳动后两颗棋子距离不变。一次只允许跳过1颗棋子。
　　
　　写一个程序，首先判断是否可以完成任务。如果可以，输出最少需要的跳动次数。

Hint

　　输入整数的绝对值均不超过$10^9$。

Solution

　　设：s1=b-a s2=c-b
　　那么b可以跳动到a左边，或者c右边。
　　如果s1 < s2，那么a可以跳到bc中间；
　　如果s1 > s2，那么c可以跳到ab中间。

　　也就是说，如果s1≠s2，那么一个局面有3种跳法。
　　如果s1=s2，那么只有2种跳法。

　　如果我们用图来表示状态之间的关系，就很容易发现，状态之间组成的联系实际上是二叉树组成的森林。
　　每一个s1=s2的状态（注意，它们的位置可能不同）都是一棵二叉树的根。
　　其余的每个状态，a或c往中间跳表示往父亲节点走一步
　　对于所有状态，中间节点往左右跳分别对应往左右孩子走一步。

　　原问题转换成了树上最短路问题。我们设起始状态对应节点p，目标状态对应节点q。那么问题是：1.p和q是否同根。2.如果同根，求p到q的距离。

　　这两个问题都可以用LCA的知识来解决。（LCA最近公共祖先）

　　如果p和q不存在LCA那么输出NO。
　　如果存在，那么计算LCA(p,q)到p和q分别的距离，相加即为答案。

　　当我们在把某状态往父亲走时，每走一步，a或c就会往b的方向跳一次。这其实相当于a或c去了b的位置，而b向右跳了s1格或向左跳了s2格。跳完之后，s2就会递减一个s1（或s1递减一个s2）。

　　于是我们思考，可不可以一次走很多步。
　　可以，简单的数学运算就能解决问题。
　　事实上我们可以在$O(log_2S)$的时间内做到把某状态往父亲走任意多步。
　　（相当于s1和s2辗转相除，已证明此操作复杂度为对数阶）

　　通过欧几里得算法，我们可以直接计算p和q在二叉树中的深度。
　　为了方便求LCA，我们首先把p和q深度调整到相同。h(x)表示x的深度。
不妨设h(p)≤h(q)。我们把q往上走h(q)-h(p)步。

　　求2个深度相同的点的LCA，我们可以采用二分答案的方法。
　　对于二分答案：LCA到p的距离mid，
　　如果p往上走mid和q往上走mid到达的点相同，
　　那么 答案≤mid；
　　否则 答案＞mid。
　
　　时间复杂度：$O((log_2S)^2)$。

Code

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
#define fo(i,a,b) for(i=a;i<=b;i++)
int i,a[3],b[3],c[3],d[3],d1,d2,l,r,mid;
int getdeep(int x,int y)
{
    return !y?0:getdeep(y,x%y)+x/y;
}
void up(int &a,int &b,int &c,int x)
{
    int s1=b-a,s2=c-b,bs;
    if(s1==s2)return;
    if(s1<s2)
    {
        bs=min((s2-1)/s1,x);
        a+=s1*bs;
        b=a+s1;
    }
    else
    {
        bs=min((s1-1)/s2,x);
        b-=s2*bs;
        c=b+s2;
    }
    if(bs<x)up(a,b,c,x-bs);
}
int main()
{
    fo(i,0,2)scanf("%d",&a[i]);
    sort(a,a+3);
    fo(i,0,2)scanf("%d",&b[i]);
    sort(b,b+3);
    d1=getdeep(a[1]-a[0],a[2]-a[1]);
    d2=getdeep(b[1]-b[0],b[2]-b[1]);
    if(d1>d2)swap(a,b),swap(d1,d2);
    up(b[0],b[1],b[2],d2-d1);
    r=d1-1;
    while(l<=r)
    {
        mid=(l+r)/2;
        memcpy(c,a,sizeof(a));
        memcpy(d,b,sizeof(b));
        up(c[0],c[1],c[2],mid);
        up(d[0],d[1],d[2],mid);
        if(l==r)break;
        if(c[0]==d[0]&&c[1]==d[1]&&c[2]==d[2])
                r=mid;
        else    l=mid+1;
    }
    if(c[0]!=d[0]||c[1]!=d[1]||c[2]!=d[2])
            printf("NO");
    else    printf("YES\n%d",d2-d1+l*2);
}