【力扣】5429. 数组中的 k 个最强值

求助！！！

以下是 力扣 第 192 场周赛 第二题的解题，感觉自己思路没错，若有不正确的思路请大佬们指出，可以优化的地方请大佬们帮帮忙，谢谢蟹蟹！

5429. 数组中的 k 个最强值 题目难度Medium

给你一个整数数组 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

提交记录

class Solution {
    public int[] getStrongest(int[] arr, int k) {
        // 1. 排序
        sortArr(arr, 0, arr.length - 1);
        
        // 2. 取中位数
        int mid = arr[(arr.length - 1) / 2];
        
        // 3. 比较
        com(arr, mid);
        
        // 4. 取前 k 个值
        return Arrays.copyOf(arr, k);
    }
    
    // 比较
    public void com(int[] arr, int mid) {
        for(int i = arr.length - 1; i > 0; i--) {
            for(int j = arr.length - 1; j > arr.length - 1 - i; j--) {
                if(comNum(arr[j], arr[j-1], mid)) {
                    int tmp = arr[j];
                    arr[j] = arr[j-1];
                    arr[j-1] = tmp;
                }
            }
        }
    }
    
    // 判断强值
    public boolean comNum(int n1, int n2, int mid) {
        // |arr[i] - m| > |arr[j] - m| 
        // |arr[i] - m| == |arr[j] - m|，且 arr[i] > arr[j]
        int v1 = n1 > mid ? n1 - mid : mid - n1;
        int v2 = n2 > mid ? n2 - mid : mid - n2;
        if(v1 > v2) {
            return true;
        } else if(v1 == v2 && n1 > n2) {
            return true;
        }
        return false;
        
    }
    
    // 快排
    public void sortArr(int[] arr, int low, int high) {
        if(low < high) {
            int flag = partitionSort(arr, low, high);
            sortArr(arr, low, flag - 1);
            sortArr(arr, flag + 1, high);
        }
    }
    
    public int partitionSort(int[] arr, int low, int high) {
        int tmp = arr[low];
        while(low < high) {
            while(low < high && arr[high] >= tmp) {
                high--;
            }
            arr[low] = arr[high];
            while(low < high && arr[low] <= tmp) {
                low++;
            }
            arr[high] = arr[low];
        }
        arr[low] = tmp;
        return low;
    }
    
}