🌈【LeetCode 1471. 数组中的 k 个最强值 】- JavaScript(双指针+排序)

144 阅读3分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第15天,点击查看活动详情


说明:文章部分内容及图片出自网络,如有侵权请与我本人联系(主页有公众号:小攻城狮学前端)

作者:小只前端攻城狮、 主页:小只前端攻城狮的主页、 来源:掘金

GitHub:P-J27、 CSDN:PJ想做前端攻城狮

著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


【LeetCode 1471. 数组中的 k 个最强值 】- JavaScript(双指针+排序)

题目描述

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

双指针

思路分析:

首先以重要值为索引,保存对应的数字数组(可能多个数字对应同样的重要值),然后反向遍历这个数组,保存到res数组中。在这道题目中影响速度的是排序算法,我一开始写了个快排,发现遇到一个全是2的用例,超级多的2,导致排序的时候超出时间限制;然后我换成了三路快排,2的用例过了,然而其中一个用例却过不了;最后我换成了归并排序才通过。

var getStrongest = function(arr, k) {
    var res = []
    if(arr.length <= 1){
        return arr
    } 
    
    var i = 0
    var j = arr.length - 1
    arr.sort((a,b)=> {return a - b})
    var mid = Math.floor((arr.length - 1)/2)
    var m = arr[mid]
    for(let n = 0;n < arr.length;n++){
        if(Math.abs(arr[i]-m) > Math.abs(arr[j]-m)){
            res[n] = arr[i]
            i++
        }else if(Math.abs(arr[i]-m) <= Math.abs(arr[j]-m)){
            res[n] = arr[j]
            j--
        }
    }
    return res.splice(0,k)
};

注意:这道题目中的中位数与平时的中位数不同,题目中指的就是位于 ((n - 1) / 2) 的元素。只要先找到中位数,再根据距中位数的距离和自身的大小再次排序。


感谢阅读,希望能对你有所帮助,文章若有错误或者侵权,可以在评论区留言或在我的主页添加公众号联系我。

写作不易,如果觉得不错,可以「点赞」+「评论」 谢谢支持❤