1471. 数组中的 k 个最强值

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

先排序后左右指针查找比较

class Solution {
public:
    vector<int> getStrongest(vector<int>& arr, int k) {
        vector<int> res;
        sort(arr.begin(),arr.end());
        int n = arr.size();
        int mid = arr[(n-1)/2];
        int left = 0;
        int right = arr.size()-1;
        while(k>0)
        {
            if(arr[right]-mid>= mid -arr[left])
            {
                res.push_back(arr[right]);
                right--;
            }
            else
            {
                res.push_back(arr[left]);
                left++;
            }
            k--;
        }
        return res;

    }
};

C++自定义排序

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);
    }
};