输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。
-
排序
这种显而易见,直接排序,取前k个,时间复杂度:O(nlongn),空间复杂度:O(1) -
堆排序
建立一个容量为k的大根堆的优先队列。遍历一遍元素,如果队列大小<k,就直接入队,否则,让当前元素与队顶元素相比,如果队顶元素大,则出队,将当前元素入队,时间复杂度:O(nlongk), 插入容量为k的大根堆时间复杂度为O(longk), 一共遍历n个元素,空间复杂度:O(k) -
部分快排
对数组[l, r]一次快排partition过程可得到,[l, p), p, [p+1, r)三个区间,[l,p)为小于等于p的值,[p+1,r)为大于等于p的值。然后再判断p,如果p+1 == k, 找到答案,如果p+1 < k, 说明答案在[p+1, r)区间内,如果p+1 > k , 说明答案在[l, p)内。时间复杂度:平均时间复杂度为O(n),每次partition的大小为n+n/2+n/4+… = 2n,最坏时间复杂度为O(n^2), 因为每次partition都只减少一个元素,空间复杂度:O(1) -
部分冒泡排序
最外层循环K次就可以了,也就是说不用全部排序,只挑出符合提议的K个就可以。
import java.util.*;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> al = new ArrayList<Integer>();
if (k > input.length) {
return al;
}
for (int i = 0; i < k; i++) {
for (int j = 0; j < input.length - i - 1; j++) {
if (input[j] < input[j + 1]) {
int temp = input[j];
input[j] = input[j + 1];
input[j + 1] = temp;
}
}
al.add(input[input.length - i - 1]);
}
return al;
}
}