重新排列前N个数字，使其距离为K

有一个长度为N的数字序列，要求重新排列后，将相邻两个元素的差的绝对值最大化，且相邻两个元素的距离为K。请编写一个函数，实现该功能。

示例

输入:

arr = [1, 2, 3, 4, 5]
k = 3

输出:

[5, 2, 4, 1, 3]

思路

首先，需要对原始序列进行排序。然后，从排完序后的数组的最后一个元素开始，每隔k个元素取一个，直到最后一个元素。

代码实现

def rearrange(arr, k):
    # 排序
    arr = sorted(arr)
    # 元素个数
    n = len(arr)
    # 创建一个新数组，用于存放重新排列后的元素
    new_arr = [0] * n
    # 从排完序后的数组的最后一个元素开始
    idx = n - 1
    # 每隔k个元素取一个，直到最后一个元素
    while idx >= 0:
        new_arr[idx] = arr.pop()
        idx -= k
    # 返回新数组
    return new_arr

测试

我们使用示例中提供的输入进行测试：

arr = [1, 2, 3, 4, 5]
k = 3

print(rearrange(arr, k))  # 输出 [5, 2, 4, 1, 3]