求助！！！
以下是 力扣 第 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;
}
}