最小的四个数
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路
第一种就是调用sort函数,然后调用数组的slice方法,很简单。
第二种是利用快速排序的原理解决问题。基于数组的第k个数字来调整,使得比第k个数字晓得所有数字都位于其左边,比第k个数字大的所有数字都位于数组的右边。调整之后,位于数组左边的k个数字就是最小的k个数字(这k个数字不一定是排序的)。
但是第二种在运行的时候超时了,不建议用。
代码
function GetLeastNumbers_Solution(input, k)
{
// write code here
input.sort();
if(input.length < k ){
return 0;
}
return input.slice(0,k);
}
//第二种(这种超时了)
function swap(arr, i, j) {
var temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
function partation(arr, start, end) {
var pivot = arr[end]; //设置哨兵
var i = start; //交换的次数+1 哨兵要在数组插入的位置
for (var j = start; j < end; j++) {
if (arr[j] < pivot) {
swap(arr, i, j);
i++;
}
}
swap(arr, i, end);
return i;
}
function GetLeastNumbers_Solution(input, k) {
var result = [];
if (input.length < 0 || k > input.length || k <= 0) {
return false;
}
var start = 0;
var end = input.length - 1;
var p = partation(input, start, end);
while (p !== (k - 1)) {
if (p > (k - 1)) {
end = p - 1;
p = partation(input, start, end);
} else {
start = p + 1;
partation(input, start, end);
}
}
for (var i = 0; i < k; i++) {
result.push(input[i]);
}
return result;
}