[bzoj2144]跳跳棋 LCA

2144: 跳跳棋

Time Limit: 10 Sec   Memory Limit: 259 MB
[ Submit][ Status][ Discuss]

Description

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

Input

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

Output

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

Sample Input

1 2 3
0 3 5

Sample Output

YES
2

【范围】
100% 绝对值不超过10^9

HINT

Source

orz....

#include<algorithm>
#include<iostream>
#include<cstdio>
#define INF 1<<30
using namespace std;
int a[5],b[5],dep,d1,d2,ans,l,r;
struct data{ int a[5]; };
data anc( int *a, int k ){
	data ans; for( int i = 1; i <= 3; i++ ) ans.a[i] = a[i];
	int c1 = a[2] - a[1], c2 = a[3] - a[2];
	if( c1 == c2 ) return ans;
	if( c1 < c2 ){
		int t = min( k, (c2-1)/c1 );
		k -= t; dep += t;
		ans.a[2] += t*c1; ans.a[1] += t*c1;
	} else {
		int t = min( k, (c1-1)/c2 );
		k -= t; dep += t;
		ans.a[3] -= t*c2; ans.a[2] -= t*c2;
	}
	if( k ) return anc(ans.a,k);
	return ans;
}
bool cmp( data a, data b ){
	for( int i = 1; i <= 3; i++ ) if( a.a[i] != b.a[i] ) return false;
	return true;
}
int main(){
	for( int i = 1; i <= 3; i++ ) scanf("%d", &a[i]); sort(a+1,a+4);
	for( int i = 1; i <= 3; i++ ) scanf("%d", &b[i]); sort(b+1,b+4);
	data x = anc( a, INF ); d1 = dep; dep = 0;
	data y = anc( b, INF ); d2 = dep; dep = 0;
	if( !cmp(x,y) ){ puts("NO"); return 0; }
	if( d1 > d2 ){ swap(d1,d2); swap(a,b); }
	ans = d2-d1; x = anc( b, ans ); for( int i = 1; i <= 3; i++ ) b[i] = x.a[i];
	l = 0; r = d1;
	while( l <= r ){
		int mid = ( l + r ) >> 1;
		if( !cmp(anc( a, mid ),anc( b, mid )) ) l = mid+1;
		else r = mid-1;
	}
	puts("YES");
	ans += l*2; printf("%d", ans);
	return 0;
}