题目描述:
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4 。
时间限制:1秒
空间限制:32768k
(http://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf?rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking)
在牛客网给出的测试数据中,假如k的值大于数组的长度,将会返回一个空链表,比较坑爹。。。
- 利用冒泡排序实现。冒泡排序是将两个相邻的元素进行比较,将比较小的那一个移动到前边去,这样,每一趟循环就可以得到一个最小的数,进行n次循环可以得到一个从小到大排列的序列。由于我们只需要最小的k个数,因此不需要进行n次循环,只需要进行k次循环。
public ArrayList<Integer> getLeastNum(int[] input, int k) {
ArrayList<Integer> list = new ArrayList();
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;
}
}
list.add(input[input.length - i - 1]);
}
return list;
}
2.将输入的前k个数存入一个大小为k的数组,循环将后面n-k个数与这个数组中最大的数进行替换,最终这个数组的数为最小的k个数。这种方法得到的数组不一定是有序的。
public ArrayList<Integer> getLeastNumbers(int[] input, int k) {
ArrayList<Integer> list = new ArrayList<Integer>();
if (input.length < k || k == 0)
return list;
for (int i = 0; i < k; i++)
list.add(input[i]);
for (int i = k; i < input.length; i++) {
int j = getMax(list);
int temp = (Integer) list.get(j);
if (input[i] < temp)
list.set(j, input[i]);
}
return list;
}
public int getMax(ArrayList<Integer> list) {
int max = list.get(0);
int j = 0;
for (int i = 0; i < list.size(); i++) {
if (list.get(i) > max) {
max = list.get(i);
j = i;
}
}
return j;
}
3.基于这个数组中第k个数进行调整,使得比第k个数小的数全在数组的左边,比第k个数大的数全在数组的右边,这样,数组的前k个数即为最小的k个数。