数组中的 k 个最强值

/*
 * 思路：
 * 1. 先对数组进行排序，然后找到中位数。
 * 2. 使用Java Streams进行强度计算和排序。
 * 3. 选择前k个元素。
 */
import java.util.*;
import java.util.stream.*;

public class Solution {
    public int[] getStrongest(int[] arr, int k) {
        // 排序数组以找到中位数
        Arrays.sort(arr);
        int n = arr.length;
        int median = arr[(n - 1) / 2];

        // 使用Java Streams进行排序和选择
        return Arrays.stream(arr)
                .boxed()
                .sorted((a, b) -> {
                    int diffA = Math.abs(a - median);
                    int diffB = Math.abs(b - median);
                    if (diffA == diffB) return b - a;
                    return diffB - diffA;
                })
                .limit(k)
                .mapToInt(i -> i)
                .toArray();
    }
}