-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathSolution.java
More file actions
45 lines (42 loc) · 1.11 KB
/
Copy pathSolution.java
File metadata and controls
45 lines (42 loc) · 1.11 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
class Solution
{
public int[] getStrongest(int[] arr, int k)
{
Arrays.sort(arr);
final int n = arr.length;
final int m = (n - 1) / 2;
final int median = arr[m];
List<Integer> nums = new ArrayList<>();
for (int num : arr)
{
nums.add(num);
}
int[] result = new int[k];
var numsAsList = nums.stream().sorted((a, b) ->
{
if (!a.equals(b))
{
int diffA = Math.abs(a - median);
int diffB = Math.abs(b - median);
if (diffA != diffB)
{
return Integer.compare(diffB, diffA);
}
else
{
return Integer.compare(b, a);
}
}
return 0;
}).limit(k).collect(Collectors.toList());
for (int i = 0; i < k; i++)
{
result[i] = numsAsList.get(i);
}
return result;
}
}