1. 问题描述:
给你一个整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/the-k-strongest-values-in-an-array
2. 思路分析:
① 首先需要理解清楚题目,一个比较简单的方法是计算出所有的数组元素与中位数之间的差值,然后将差值相同的元素值放入到map中,因为最后是要求解出最强的k个元素,所以我们可以使用treemap来实现map接口,这样可以自定义键为大到小的顺序,这样我们在遍历的时候是从差值比较大的元素开始的,值为List<Integer>可以将所有的差值相同的连接在一起
② 因为放入List的时候有可能不是从大到小的顺序排列的,所以我们对遍历到的List进行排序,然后放入到结果数组即可
3. 代码如下:
class Solution {
/*想到使用Treemap来进行映射, 将相同的距离的数组值放在一起*/
public int[] getStrongest(int[] arr, int k) {
Arrays.sort(arr);
Map<Integer, List<Integer>> map = new TreeMap<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2 - o1;
}
});
int mid = (arr.length - 1) / 2;
for (int i = 0; i < arr.length; ++i){
List<Integer> cur = map.getOrDefault(Math.abs(arr[i] - arr[mid]), new ArrayList<>());
cur.add(arr[i]);
map.put(Math.abs(arr[i] - arr[mid]), cur);
}
int []res = new int[k];
int n = 0;
for (Map.Entry<Integer, List<Integer>> entry : map.entrySet()){
List<Integer> cur = entry.getValue();
Collections.sort(cur);
for (int i = cur.size() - 1; i >= 0; --i) {
if (n == k) return res;
res[n++] = cur.get(i);
}
}
return res;
}
}