var getLeastNumbers = function(arr, k) {
let len = arr.length
if (!len || !k) return []
let start = 0
let end = len - 1
// 寻找一次标杆元素的位置
let index = quikSort(arr, start, end)
// 如果标杆元素的位置不等于 K
while(index !== k - 1) {
if (index > k-1) {
// 如果上一次查找,标杆元素位置大于目标位置
end = index-1
index = quikSort(arr, start, end)
} else {
// 如果上一次查找,标杆元素位置小于目标位置
start = index + 1
index = quikSort(arr, start, end)
}
}
return arr.slice(0, index+1)// 在index+1处结束但不包括
};
function quikSort(arr, left, right) {
let pivot = arr[left]
while(left < right) {
while(left < right && arr[right] >= pivot) right--
arr[left] = arr[right]
while(left < right && arr[left] < pivot) left++
arr[right] = arr[left]
}
arr[left] = pivot
return left
}
package com.nrsc.sort;
public class QuickSort {
public static void main(String[] args) {
int[] arr = { 49, 38, 65, 97, 23, 22, 76, 1, 5, 8, 2, 0, -1, 22 };
quickSort(arr, 0, arr.length - 1);
System.out.println("排序后:");
for (int i : arr) {
System.out.println(i);
}
}
private static void quickSort(int[] arr, int low, int high) {
if (low < high) {
// 找寻基准数据的正确索引
int index = getIndex(arr, low, high);
// 进行迭代对index之前和之后的数组进行相同的操作使整个数组变成有序
//quickSort(arr, 0, index - 1); 之前的版本,这种姿势有很大的性能问题,谢谢大家的建议
quickSort(arr, low, index - 1);
quickSort(arr, index + 1, high);
}
}
private static int getIndex(int[] arr, int low, int high) {
// 基准数据
int tmp = arr[low];
while (low < high) {
// 当队尾的元素大于等于基准数据时,向前挪动high指针
while (low < high && arr[high] >= tmp) {
high--;
}
// 如果队尾元素小于tmp了,需要将其赋值给low
arr[low] = arr[high];
// 当队首元素小于等于tmp时,向前挪动low指针
while (low < high && arr[low] <= tmp) {
low++;
}
// 当队首元素大于tmp时,需要将其赋值给high
arr[high] = arr[low];
}
// 跳出循环时low和high相等,此时的low或high就是tmp的正确索引位置
// 由原理部分可以很清楚的知道low位置的值并不是tmp,所以需要将tmp赋值给arr[low]
arr[low] = tmp;
return low; // 返回tmp的正确位置
}
}