题意

给你一个整数数组 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.length <= 10^5
• -10^5 <= arr[i] <= 10^5
• 1 <= k <= arr.length

样例

样例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] 的任何排列都是正确答案。

算法1（自定义cmp函数）

思路

先排序找到中位数

安装题目的比较规则再次排序，取排序后的前k的数即可

比较函数可在类外定义，亦可用lambda表达式

复杂度

时间：$0（ 2* nlogn ）$

c++ 代码

 int m;
 bool cmp(int a, int b)
 {
     return abs(a - m) == abs(b - m) ?  a > b : abs(a - m) > abs(b - m);
 }

class Solution {
public:
    vector<int> getStrongest(vector<int>& arr, int k) {
        sort(arr.begin(),arr.end());
        m = arr[(arr.size() - 1 )/ 2];

        sort(arr.begin(), arr.end(), cmp);

        return vector<int> (arr.begin(), arr.begin() + k);
    }

};

// -------lambda -------------------------------------------
class Solution {
public:
    vector<int> getStrongest(vector<int>& arr, int k) {
        sort(arr.begin(),arr.end());
        int n=arr.size();
        int m=arr[(n-1)/2];

        sort(arr.begin(),arr.end(),
        [&](int a, int b){int x = abs( a - m);int y = abs(b - m);return x == y ? a > b : x > y;});
        // 要传参数
        return vector<int>(arr.begin(),arr.begin()+k);

    }
};

算法2 （排序 + 双指针）

思路

先排序，找到中位数m。

然后首尾双指针比较 |arr[l] - m| >= |arr[l] - m|

注意： 排序后其实将比较条件中的 |arr[i] - m| == |arr[j] - m|，且 arr[i] > arr[j] 和 |arr[i] - m| > |arr[j] - m| 用 |arr[l] - m| >= |arr[l] - m| 统一来表示

复杂度

时间 ：$O(nlogn + n)$

c++ 代码

class Solution {
public:
    vector<int> getStrongest(vector<int>& arr, int k) {
        sort(arr.begin(),arr.end());
        int l = 0 , r = arr.size() - 1;

        int m = arr[r / 2];

        vector<int> res;
        // 这里比较时已经满足了 第一个条件 直接判断第一个规则即可
        while(k --)
        {
            if(abs(arr[r] - m) >= abs(arr[l] - m))
                res.push_back(arr[r -- ]);
            else
                res.push_back(arr[l ++ ]);
        }

        return res;
    }

};