最近在力扣刷题的时候刷到一个自己感觉比较有意义的题目:
给你一个整数数组 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 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/the-k-strongest-values-in-an-array
package lmc;
import java.util.Arrays;
public class test {
public static int[] getStrongest(int[] arr, int k) {
if (k>=arr.length||arr.length==0) {
return arr;
}
Arrays.sort(arr);
int zws = arr[(arr.length-1)/2];
int [] ans = new int [k];
int l = 0;
int r = arr.length-1;
for (int i = 0; i < ans.length; i++) {
if (Math.abs(arr[r]-zws)>=Math.abs(arr[l]-zws)) {
ans[i]=arr[r];
r--;
}else{
ans[i]=arr[l];
l++;
}
}
return ans ;
}
public static void main(String[] args) {
int [] arr={6,7,11,7,6,8};
int k = 5;
getStrongest(arr,k);
}
}