Leetcode 5429 数组中的 k 个最强值

1.题目链表

https://leetcode-cn.com/problems/the-k-strongest-values-in-an-array/

2.题目链接

给你一个整数数组 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| ,但是 51 更强,因为 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

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

3.题目解析

1)根据题目描述,首先找出数组的中位数m,用qsort函数对数组进行从小到大排序,中位数即为(arrSize - 1) / 2下标的数;

2)用左右指针实现数组的遍历,比较abs(arr[left] - m)abs(arr[right] - m)的大小,即可找出满足条件的数;

4.代码实现

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int comp(const void *a, const void *b) {
    return *(int *)a - *(int *)b;
}

int* getStrongest(int* arr, int arrSize, int k, int* returnSize){
    qsort(arr, arrSize, sizeof(int), comp);
    int m = arr[(arrSize - 1) / 2];
    int *res = (int *)calloc(k, sizeof(int));
    int left = 0;
    int right = arrSize - 1;
    *returnSize = k;
    int cnt = 0;

    while (left < right) {
        if(abs(arr[left] - m) > abs(arr[right] - m))
            res[cnt++] = arr[left++];
        else
            res[cnt++] = arr[right--];
        if (cnt == k)
            break;
    }
    if (cnt == k - 1) 
        res[cnt++] = arr[left];
    return res;
}

5.提交记录

stay hungry, stay foolish
原文地址:https://www.cnblogs.com/zygote/p/13062681.html