[JavaScript]寻找第K大

描述

有一个整数数组，请你根据快速排序的思路，找出数组中第k 大的数。

给定一个整数数组 ,同时给定它的大小n和要找的 ，请返回第k 大的数(包括重复的元素，不用去重)，保证答案存在。
要求时间复杂度 O(n)

示例1

输入：
[1,3,5,2,2],5,3
返回值：
2

示例2

输入：
[10,10,9,9,8,7,5,6,4,3,4,2],12,3
返回值：
9
说明：
去重后的第3大是8，但本题要求包含重复的元素，不用去重，所以输出9

代码

sort slice解法

/**
 * 
 * @param a int整型一维数组 
 * @param n int整型 
 * @param K int整型 
 * @return int整型
 */
function findKth( a ,  n ,  K ) {
    // write code here
    a.sort((x,y)=>{
        return x-y
    })
    return a.splice((n-K),1)
}
module.exports = {
    findKth : findKth
};

经典快排

/**
 * 
 * @param a int整型一维数组 
 * @param n int整型 
 * @param K int整型 
 * @return int整型
 */
function findKth( a ,  n ,  K ) {
    // write code here
    return quick(a,0,a.length-1).splice((n-K),1)
}
function quick(arr,left,right){
    if(arr.length>1){
        var index=partition(arr,left,right)
        if(left<index-1){
            quick(arr,left,index-1)
        }
        if(index<right){
            quick(arr,index,right)
        }
    }
    return arr
}
function partition(arr,left,right){
    var i=left,j=right,goal=arr[left]
    while(i<=j){
        while(arr[i]<goal){
            i++
        }
        while(arr[j]>goal){
            j--
        }        
        if(i<=j){
            [arr[i],arr[j]]=[arr[j],arr[i]]
            i++
            j--
        }
    }
    return i
}
module.exports = {
    findKth : findKth
};