[路飞]_leetcode_面试题 17.14. 最小K个数

设计一个算法，找出数组中最小的k个数。以任意顺序返回这k个数均可。

示例：

输入： arr = [1,3,5,7,2,4,6,8], k = 4
输出： [1,2,3,4]

解题思路

这一题可以通过快排优化排序所操作的次数，然后取最小值

代码

var threshold = 16;
var swap = function(arr, i, j) {
    [arr[j], arr[i]] = [arr[i], arr[j]];
}
var getmid = function(arr, a, b, c) {
    if (arr[a] > arr[b]) swap(arr, a, b);
    if (arr[a] > arr[c]) swap(arr, a, c);
    if (arr[b] > arr[c]) swap(arr, b, c);
    return arr[b];
}
var __quick_sort = function(arr, l, r) {
    while(r - l > threshold) {
        let x = l, y = r, base = getmid(arr, l, (l + r) / 2 | 1, r);
        do{
            while(arr[x] < base) x++;
            while(arr[y] > base) y--;
            if (x <= y) {
                swap(arr, x, y)
                x++, y--;
            }
        }while(x <= y);
        __quick_sort(arr, x, r);
        r = y;
    }
}
// 插入排序
var final_insert_sort = function(arr, l, r) {
    let ind = l;
    // 找到最小值
    for (let i = l + 1; i <= r; i++) {
        if (arr[i] < arr[ind]) ind = i;
    }
    while(ind > l) {
        swap(arr, ind, ind - 1);
        ind--;
    }
    
    for(let i = l + 2; i <= r; i++) {
        let j = i;
        while(arr[j] < arr[j - 1]) {
            swap(arr, j, j - 1);
            j--;
        }
    }
}

var quick_sort = function(arr, l, r) {
    __quick_sort(arr, l, r);
    final_insert_sort(arr, l, r);
}
var smallestK = function(arr, k) {
    if (arr.length <= k) return arr;
    quick_sort(arr, 0, arr.length - 1);
    let ans = [];
    while(--k >= 0) ans.push(arr[k]);
    return ans;
};