/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
* @param a int整型一维数组
* @param n int整型
* @param K int整型
* @return int整型
*/
/**
* 解法一:快速排序
* 思路:快排,取得升序数组,然后取下标 nums.length - k + 1 对应的值
* 时间复杂度:O(nlogn)
* 空间复杂度:最优 O(logn), 最差 O(n)
*/
export function findKth(a: number[], n: number, K: number): number {
if (a.length === 0) return 0
return sortArray(a)[a.length - K + 1]
}
function sortArray(nums: number[]): number[] {
const len = nums.length
if (len === 0) return nums
const midIndex = Math.floor(len / 2)
const midValue = nums.splice(midIndex, 1)[0]
const left: number[] = []
const right: number[] = []
// 注意: splice 会修改原数组,所以用 arr.length
for (let i = 0; i < nums.length; i++) {
const n = nums[i]
if (n < midValue) {
left.push(n)
} else {
right.push(n)
}
}
return sortArray(left).concat([midValue], sortArray(right))
};
一站式解决前端面试高频算法题