华为od机考 Python 【中位数】

题目

假设你有一个正整数数组X和一个整数K。请计算下式的结果：

X[i] - X[i + 1] - … - X[i + K - 1]

你的目标是找到使这个结果与数组的中位数最为接近的i值。如果有多个相同的结果，请返回最大的i。

额外信息：

数组的中位数是什么？当你把数组从小到大排序后，中间那个数字就是中位数。
你可以假设数组X里的元素都是正整数。
数组X的长度n限制为：2 ≤ n ≤ 1000。
K的值肯定小于数组的大小。
i的范围是：0 ≤ i < 1000。
例子:
输入: [50,50,2,3], 2
输出: 1

为什么？首先，这个数组排序后为[2,3,50,50]，中位数是50。接着，我们按照上面的公式计算，找到差值与50最接近的是X[1] - X[2]，所以返回1。

代码

def closest_subarray_to_median(nums