C++刷leetcode（5429. 数组中的 k 个最强值）【数据结构】【sort】
本帖最后由 糖逗 于 2020-6-7 12:57 编辑题目描述：
给你一个整数数组 arr 和一个整数 k 。
设 m 为数组的中位数，只要满足下述两个前提之一，就可以判定 arr 的值比 arr 的值更强：
|arr - m| > |arr - m|
|arr - m| == |arr - m|，且 arr > arr
请返回由数组中最强的 k 个值组成的列表。答案可以以 任意顺序 返回。
中位数 是一个有序整数列表中处于中间位置的值。形式上，如果列表的长度为 n ，那么中位数就是该有序列表（下标从 0 开始）中位于 ((n - 1) / 2) 的元素。
例如 arr = ，n = 5：数组排序后得到 arr = [-3, 2, 6, 7, 11] ，数组的中间位置为 m = ((5 - 1) / 2) = 2 ，中位数 arr 的值为 6 。
例如 arr = [-7, 22, 17, 3]，n = 4：数组排序后得到 arr = [-7, 3, 17, 22] ，数组的中间位置为 m = ((4 - 1) / 2) = 1 ，中位数 arr 的值为 3 。
示例 1：
输入：arr = , k = 2
输出：
解释：中位数为 3，按从强到弱顺序排序后，数组变为 。最强的两个元素是 。 也是正确答案。
注意，尽管 |5 - 3| == |1 - 3| ，但是 5 比 1 更强，因为 5 > 1 。
示例 2：
输入：arr = , k = 2
输出：
解释：中位数为 3, 按从强到弱顺序排序后，数组变为 。最强的两个元素是 。
示例 3：
输入：arr = , k = 5
输出：
解释：中位数为 7, 按从强到弱顺序排序后，数组变为 。
的任何排列都是正确答案。
示例 4：
输入：arr = , k = 3
输出：[-3,11,2]
示例 5：
输入：arr = [-7,22,17,3], k = 2
输出：
提示：
1 <= arr.length <= 10^5
-10^5 <= arr <= 10^5
1 <= k <= arr.length
class Solution {
public:
vector<int> getStrongest(vector<int>& arr, int k) {
sort(arr.begin(), arr.end());
int len = arr.size();
int median = arr[(len-1)/2];
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;});
return vector<int>(arr.begin(),arr.begin()+k);
}
};
页:
[1]