C++刷leetcode（5429. 数组中的 k 个最强值）【数据结构】【sort】

糖逗

题目描述：

1. 给你一个整数数组 arr 和一个整数 k 。
   
2. 
   
3. 设 m 为数组的中位数，只要满足下述两个前提之一，就可以判定 arr[i] 的值比 arr[j] 的值更强：
   
4. 
   
5. |arr[i] - m| > |arr[j] - m|
   
6. |arr[i] - m| == |arr[j] - m|，且 arr[i] > arr[j]
   
7. 请返回由数组中最强的 k 个值组成的列表。答案可以以 任意顺序 返回。
   
8. 
   
9. 中位数 是一个有序整数列表中处于中间位置的值。形式上，如果列表的长度为 n ，那么中位数就是该有序列表（下标从 0 开始）中位于 ((n - 1) / 2) 的元素。
   
10. 
    
11. 例如 arr = [6, -3, 7, 2, 11]，n = 5：数组排序后得到 arr = [-3, 2, 6, 7, 11] ，数组的中间位置为 m = ((5 - 1) / 2) = 2 ，中位数 arr[m] 的值为 6 。
    
12. 例如 arr = [-7, 22, 17, 3]，n = 4：数组排序后得到 arr = [-7, 3, 17, 22] ，数组的中间位置为 m = ((4 - 1) / 2) = 1 ，中位数 arr[m] 的值为 3 。
    
13. 
    
14. 
    
15. 示例 1：
    
16. 
    
17. 输入：arr = [1,2,3,4,5], k = 2
    
18. 输出：[5,1]
    
19. 解释：中位数为 3，按从强到弱顺序排序后，数组变为 [5,1,4,2,3]。最强的两个元素是 [5, 1]。[1, 5] 也是正确答案。
    
20. 注意，尽管 |5 - 3| == |1 - 3| ，但是 5 比 1 更强，因为 5 > 1 。
    
21. 示例 2：
    
22. 
    
23. 输入：arr = [1,1,3,5,5], k = 2
    
24. 输出：[5,5]
    
25. 解释：中位数为 3, 按从强到弱顺序排序后，数组变为 [5,5,1,1,3]。最强的两个元素是 [5, 5]。
    
26. 示例 3：
    
27. 
    
28. 输入：arr = [6,7,11,7,6,8], k = 5
    
29. 输出：[11,8,6,6,7]
    
30. 解释：中位数为 7, 按从强到弱顺序排序后，数组变为 [11,8,6,6,7,7]。
    
31. [11,8,6,6,7] 的任何排列都是正确答案。
    
32. 示例 4：
    
33. 
    
34. 输入：arr = [6,-3,7,2,11], k = 3
    
35. 输出：[-3,11,2]
    
36. 示例 5：
    
37. 
    
38. 输入：arr = [-7,22,17,3], k = 2
    
39. 输出：[22,17]
    
40. 
    
41. 
    
42. 提示：
    
43. 
    
44. 1 <= arr.length <= 10^5
    
45. -10^5 <= arr[i] <= 10^5
    
46. 1 <= k <= arr.length

复制代码

1. class Solution {
   
2. public:
   
3.     vector<int> getStrongest(vector<int>& arr, int k) {
   
4.         sort(arr.begin(), arr.end());
   
5.         int len = arr.size();
   
6.         int median = arr[(len-1)/2];
   
7.         sort(arr.begin(), arr.end(), [&](int&a, int&b){int x = abs(a - median); int y = abs(b - median); return x == y ? a > b : x > y;});
   
8.         return vector<int>(arr.begin(),arr.begin()+k);
   
9.     }
   
10. };

复制代码