题目:输入n个整数,找出其中最小的k个数。例如输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
来源于——《剑指offer》
- 思路1:利用快速排序的
patition
函数, 先随机找到一个中间数字,返回索引值,通过比较索引值和k的大小来判断是否再次调用patition
函数,知道k和返回的索引相同。 - 缺点:修改了原来的数组
public class Solution {
private void swap(int[] array, int i, int j) {
int temp = array[i];
array[i] = array[j];
array[j] = temp;
}
private int partition(int[] array, int start, int end) {
int mid = (start + end) >> 1;
if (array[start] > array[end]) {
swap(array, start, end);
}
if (array[mid] > array[end]) {
swap(array, mid, end);
}
if (array[mid] > array[start]) {
swap(array, mid, start);
}
int key = array[start];
while (start < end) {
while (start < end && key <= array[end]) {
--end;
}
array[start] = array[end];
while (start < end && key >= array[start]) {
++start;
}
array[end] = array[start];
}
array[start] = key;
return start;
}
public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
ArrayList<Integer> resList = new ArrayList<>();
if (input == null || input.length == 0 || k <= 0 || k > input.length) {
return resList;
}
int start = 0;
int end = input.length - 1;
int index = partition(input, start, end);
while (index != k - 1) {
if (index > k - 1) {
end = index - 1;
index = partition(input, start, end);
} else {
start = index + 1;
index = partition(input, start, end);
}
}
for (int i = 0; i < k; i++) {
resList.add(input[i]);
}
return resList;
}
}
- 思路2:利用红黑树,先将前k个数装入set中,然后用set中最大值与后面的数比较,如果找到比最大数小的数就将最大数移除,装入当前数,直到数组中的数遍历完毕。
public class Solution2 {
public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
ArrayList<Integer> resList = new ArrayList<>();
if (input == null || input.length == 0 || k <= 0 || k > input.length) {
return resList;
}
TreeSet<Integer> set = new TreeSet<>();
for (int i = 0; i < input.length; i++) {
if (i < k) {
set.add(input[i]);
} else {
if (input[i] < set.last()) {
set.remove(set.pollLast());
set.add(input[i]);
}
}
}
for (Integer integer : set) {
resList.add(integer);
}
return resList;
}
}