华为OD机试 Python 【计算最接近的数】

描述

你手里拿着一副扑克，你要找到连续K张牌的组合，它们之间的点数差距最接近中间那张牌。

给你一堆数字组成的数组X和一个数字K。你要找出连续K个数字，使它们的组合差最接近这堆数字的中间数。如果有多个组合，那就给我最后一个组合的起始位置。

例如，数组是 [50,50,2,3]，K是2。那么最接近的组合是 50 和 2，所以答案是1（因为50是从索引1开始的）。

小贴士：中间数，就是把数组从小到大排列后，中间的那个数字。

限制
数组X的元素全部是正整数。
X的长度是在2到1000之间。
K是大于0但小于数组的长度。

输入
[50,50,2,3],2
输出
1

代码

def find_best_index(arr, K):
    # 先求中位数
    sorted_arr = sorted(arr