蜂巢（蓝桥杯python）

问题描述

蜂巢由大量的六边形拼接而成, 定义蜂巢中的方向为: 0 表示正西方向, 1 表示西偏北 60°,2 表示东偏北 60°,3 表示正东, 4 表示东偏南 60°,5 表示西偏南 60°。

对于给定的一点 O, 我们以 O 为原点定义坐标系, 如果一个点 A 由 O 点 先向 d 方向走 p 步再向 (d+2) mod 6方向 ( d 的顺时针 120° 方向) 走 q 步到 达, 则这个点的坐标定义为 (d,p,q) 。在蜂窝中, 一个点的坐标可能有多种。

下图给出了点 B(0,5,3) 和点C(2,3,2) 的示意。

给定点 (d1,p1,q1) 和点 (d2,p2,q2), 请问他们之间最少走多少步可以到达?

输入格式

输入一行包含 6 个整数 d1,p1,q1,d2,p2,q2 表示两个点的坐标, 相邻两个整数之间使用一个空格分隔。

输出格式

输出一行包含一个整数表示两点之间最少走多少步可以到达。

样例输入

0 5 3 2 3 2

样例输出

7

评测用例规模与约定

对于 25% 的评测用例, p1,p2≤10^3;

对于 50% 的评测用例, p1,p2≤10^5;

对于 75% 的评测用例, p1,p2≤10^7;

对于所有评测用例, 0≤d1,d2≤5, 0≤q1<p1≤10^9, 0≤q2<p2≤10^9 。

运行限制

• 最大运行时间：1s

• 最大运行内存: 512M

解析

1. 将原来的3位坐标转为直角坐标系方便运算

0-5方向与O点相邻的坐标可转换为direction = [(-2, 0), (-1, 1), (1, 1), (2, 0), (1, -1), (-1, -1)]，坐标轴上的点和非坐标轴上的点到O点的距离不一样没关系，纵坐标设为1方便算步数（如果设为根号3算步数时还需除以根号3）。

坐标转换公式为：

x = direction[d][0] * p + direction[(d + 2) % 6][0] * q

y = direction[d][1] * p + direction[(d + 2) % 6][1] * q

2. B(x1, y1)与C(x2, y2)的关系分为5种：

1）B、C在同一行：y1=y2

最简单，沿方向0或3移动即可，移动步数：|x2-x1|//2

2）B在C的左上：y1>y2且x1<=x2

①先把B沿方向4移动到B'，使得B'与C同行，再把B'沿方向3移动到C；

②但如果B'到了C点右边，则会使得移动的步数不是最少，此时要先把B沿方向4移动到C点正上方的B'，再把B'移动到C。

以下三种情况类似。

3）B在C的左下：y1<y2且x1<=x2

4）B在C的右上：y1>y2且x1>x2

5）B在C的右下：y1<y2且x1>x2

代码

direction = [(-2, 0), (-1, 1), (1, 1), (2, 0), (1, -1), (-1, -1)]


def transform(d, p, q):  # 把坐标转为直角坐标系下的值
    # 设蜂巢正六边形中心到各边的距离=1
    x = direction[d][0] * p + direction[(d + 2) % 6][0] * q
    y = direction[d][1] * p + direction[(d + 2) % 6][1] * q
    return x, y


def count_step(d1, p1, q1, d2, p2, q2):
    x1, y1 = transform(d1, p1, q1)  # B点坐标
    x2, y2 = transform(d2, p2, q2)  # C点坐标

    y_steps = abs(y1 - y2)

    # 1. B点在C点同一行的左边或右边
    if y1 == y2:
        return abs(x2 - x1) // 2

    # 2. B点在C点左上，从B点出发先沿方向4走到与C同行，再沿方向3到C点
    if y1 > y2 and x1 <= x2:
        x11 = x1 + y_steps
        if x11 <= x2:  # B点到与C点同一行的位置，且B在C的左边
            x_steps = (x2 - x11) // 2
            return y_steps + x_steps
        else:
            # 先沿方向4走 x2-x1 步，走到与C点横坐标相等的位置
            y_steps_1 = x2 - x1
            # 此时在纵轴，走了 y_steps1 步
            y1 = y1 - y_steps_1
            # 还需要从C点正上方向左向右折线走到C点
            y_steps_2 = y1 - y2
            return y_steps_1 + y_steps_2

    # 3. B点在C点左下，从B点出发先沿方向2走到与C同行，再沿方向3到C点
    if y1 < y2 and x1 <= x2:
        x11 = x1 + y_steps
        if x11 <= x2:
            x_steps = (x2 - x11) // 2
            return y_steps + x_steps
        else:
            y_steps_1 = x2 - x1
            y1 = y1 + y_steps_1
            y_steps_2 = y2 - y1
            return y_steps_1 + y_steps_2

    # 4. B点在C点右上，从B点出发先沿方向5走到与C同行，再沿方向0到C点
    if y1 > y2 and x1 >= x2:
        x11 = x1 - y_steps
        if x11 >= x2:
            x_steps = (x11 - x2) // 2
            return y_steps + x_steps
        else:
            y_steps_1 = x1 - x2
            y1 = y1 - y_steps_1
            y_steps_2 = y1 - y2
            return y_steps_1 + y_steps_2

    # 5. B点在C点右下，从B点出发先沿方向1走到与C同行，再沿方向0到C点
    if y1 < y2 and x1 >= x2:
        x11 = x1 - y_steps
        if x11 >= x2:
            x_steps = (x11 - x2) // 2
            return y_steps + x_steps
        else:
            y_steps_1 = x1 - x2
            y1 = y1 + y_steps_1
            y_steps_2 = y2 - y1
            return y_steps_1 + y_steps_2


d1, p1, q1, d2, p2, q2 = map(int, input().split())
print(int(count_step(d1, p1, q1, d2, p2, q2)))