蓝桥杯：移动距离--Python解法

问题描述：

X星球居民小区的楼房全是一样的，并且按矩阵样式排列。其楼房的编号为1,2,3...
当排满一行时，从下一行相邻的楼往反方向排号。
比如：当小区排号宽度为6时，开始情形如下：

1 2 3 4 5 6
12 11 10 9 8 7
13 14 15 .....

我们的问题是：已知了两个楼号m和n，需要求出它们之间的最短移动距离（不能斜线方向移动）

输入格式：

输入为3个整数w m n，空格分开，都在1到10000范围内
w为排号宽度，m,n为待计算的楼号。

输出格式：

要求输出一个整数，表示m n 两楼间最短移动距离。

例如：

用户输入：

6 8 2

程序应该输出：

4

再例如：

用户输入：

4 7 20

程序应该输出：

5

资源约定：

峰值内存消耗 < 256M
CPU消耗 < 1000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。

代码：

def count_row(w, num):
    return num // w if num % w != 0 else num // w - 1

def count_colum(w, num, r):
    if r % 2 == 0:
        colum = num % w - 1 if num % w != 0 else w - 1
    else:
        colum = w - num % w if num % w != 0 else 0
    return colum

def run():
    w, m, n = (int(i) for i in input().split())

    x1 = count_row(w, m)
    y1 = count_colum(w, m, x1)

    x2 = count_row(w, n)
    y2 = count_colum(w, n, x2)

    print(abs(x1 - x2) + abs(y1 - y2))

run()