【LeetCode】1471. 數組中的 k 個最強值 The k Strongest Values in an Array (Python)

• 作者： 負雪明燭
• id： fuxuemingzhu
• 個人博客：http://fuxuemingzhu.cn/

---

題目地址：https://leetcode-cn.com/problems/the-k-strongest-values-in-an-array/

題目描述

給你一個整數數組 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. 1 <= arr.length <= 10^5
2. -10^5 <= arr[i] <= 10^5
3. 1 <= k <= arr.length

題目大意

找出數組中 Top K 個數字，判斷方式是各個數字與 中位數 的距離。

解題方法

自定義排序

本題需要注意的是按照題目所說去找中位數。這個中位數就是數組排序後的 arr[(N - 1) // 2]。

然後需要第二次排序，排序需要自定義 cmp 來比較與中位數的距離。

Python2 的自定義比較函數可以用 cmp = 函數，該比較函數是按照題目所說的比較規則。

最後需要注意的是排序後的 Top K 的位置在哪裏，切片時不要切錯。

Python 代碼如下：

class Solution(object):
    def getStrongest(self, arr, k):
        """
        :type arr: List[int]
        :type k: int
        :rtype: List[int]
        """
        N = len(arr)
        arr.sort()
        m = arr[(N - 1) // 2]
        def myCmp(x, y):
            if abs(x - m) > abs(y - m) or (abs(x - m) == abs(y - m) and x > y):
                return 1
            else:
                return -1
        arr.sort(cmp = myCmp, reverse = True)
        return arr[:k]

歡迎關注負雪明燭的刷題博客，leetcode刷題800多，每道都講解了詳細寫法！

日期

2020 年 6 月 21 日 —— 今晚爭取也直播講題