LeetCode-Python-1471. 数组中的 k 个最强值（数组 + 自定义排序 + 桶排序 + 双指针）

给你一个整数数组 arr 和一个整数 k 。

设 m 为数组的中位数，只要满足下述两个前提之一，就可以判定 arr[i] 的值比 arr[j] 的值更强：

 |arr[i] - m| > |arr[j] - m|
 |arr[i] - m| == |arr[j] - m|，且 arr[i] > arr[j]
请返回由数组中最强的 k 个值组成的列表。答案可以以 任意顺序 返回。

中位数 是一个有序整数列表中处于中间位置的值。形式上，如果列表的长度为 n ，那么中位数就是该有序列表（下标从 0 开始）中位于 ((n - 1) / 2) 的元素。

例如 arr = [6, -3, 7, 2, 11]，n = 5：数组排序后得到 arr = [-3, 2, 6, 7, 11] ，数组的中间位置为 m = ((5 - 1) / 2) = 2 ，中位数 arr[m] 的值为 6 。
例如 arr = [-7, 22, 17, 3]，n = 4：数组排序后得到 arr = [-7, 3, 17, 22] ，数组的中间位置为 m = ((4 - 1) / 2) = 1 ，中位数 arr[m] 的值为 3 。
 

示例 1：

输入：arr = [1,2,3,4,5], k = 2
输出：[5,1]
解释：中位数为 3，按从强到弱顺序排序后，数组变为 [5,1,4,2,3]。最强的两个元素是 [5, 1]。[1, 5] 也是正确答案。
注意，尽管 |5 - 3| == |1 - 3| ，但是 5 比 1 更强，因为 5 > 1 。
示例 2：

输入：arr = [1,1,3,5,5], k = 2
输出：[5,5]
解释：中位数为 3, 按从强到弱顺序排序后，数组变为 [5,5,1,1,3]。最强的两个元素是 [5, 5]。
示例 3：

输入：arr = [6,7,11,7,6,8], k = 5
输出：[11,8,6,6,7]
解释：中位数为 7, 按从强到弱顺序排序后，数组变为 [11,8,6,6,7,7]。
[11,8,6,6,7] 的任何排列都是正确答案。
示例 4：

输入：arr = [6,-3,7,2,11], k = 3
输出：[-3,11,2]
示例 5：

输入：arr = [-7,22,17,3], k = 2
输出：[22,17]
 

提示：

1 <= arr.length <= 10^5
-10^5 <= arr[i] <= 10^5
1 <= k <= arr.length

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/the-k-strongest-values-in-an-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

第一种思路：

模拟法，自定义排序然后返回前k项。

时间复杂度：O（NlogN)

空间复杂度：O（N）

class Solution(object):
    def getStrongest(self, arr, k):
        """
        :type arr: List[int]
        :type k: int
        :rtype: List[int]
        """
        arr.sort()
        m = arr[(len(arr) - 1) // 2]
        return sorted(arr, key = lambda x: abs(x - m))[-k:]

第二种思路：

题目给出数据范围，所以也可以用桶排序来实现。

首先因为有负数，所以要把所有数往右移 shift = max(0, 0 - min(arr） 个距离保证每个数都可以入桶。

其次遍历所有的桶，找到中位数（这题中位数定义跟数学上不太一样）。

然后双指针分别从最左桶和最右桶出发，往中位数夹。

时间复杂度：O（N + minVal + maxVal）， N = len(arr), minVal = min(arr), maxVal = max(arr)

空间复杂度：O（minVal + maxVal)

class Solution(object):
    def getStrongest(self, arr, k):
        """
        :type arr: List[int]
        :type k: int
        :rtype: List[int]
        """
        n, minVal, maxVal = len(arr), min(arr), max(arr)
        shift = max(0, 0 - minVal)

        nums = [0 for _ in range(maxVal + shift + 1)]
        for num in arr:
            nums[num + shift] += 1 # 把所有数字向右移动 shift
        
        cnt = -1
        for i, x in enumerate(nums): # 找中位数 m 
            if x:
                cnt += x
                if cnt >= ((n - 1) // 2):
                    m = i
                    break

        left = minVal + shift # 从左右最远端开始向中间夹
        right = maxVal + shift
        res = []
        cnt = 0
        while cnt < k:
            while left < len(nums) and not nums[left]:
                left += 1
            while right and not nums[right]:
                right -= 1

            if m - left <= right - m:
                nums[right] -= 1
                res.append(right - shift)
            else:
                nums[left] -= 1
                res.append(left - shift)
            cnt += 1 
            
        return res