BZOJ2144跳跳棋——LCA+二分

题目描述

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

BZOJ2144跳跳棋——LCA+二分

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

输入

第一行包含三个整数,表示当前棋子的位置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\n");
printf("%lld",len+ans*2);
}
上一篇:c/c++ 数据结构 链表插入数据代码(二)


下一篇:Java中1.0 / 0.0 会输出什么?