1. 题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
2. 解题思路
(1)可以先排序数组,然后再把前n个数字放到集合中,返回集合即可。
(2)也可以用堆的思想去做,先将前K个数放入数组,进行堆排序,若之后的数比它还小,则进行调整。
3. 代码实现
(1)
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> arrayList = new ArrayList<>();
if(input.length < k || input == null)
return arrayList;
int temp = 0;
for(int i = 0; i < input.length; i++){
for (int j = i+1; j < input.length; j++){
if (input[i] > input[j])
{
temp = input[i];
input[i] = input[j];
input[j] = temp;
}
}
if(i < k)
arrayList.add(input[i]);
if (i == k)
break;
}
return arrayList;
}
}
(2)
import java.util.ArrayList;
import java.util.Arrays;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<>();
if (input == null || k <= 0 || k > input.length)
return list;
int[] kArray = Arrays.copyOfRange(input, 0, k);
//创建大根堆
buildHeap(kArray);
for (int i = k; i < input.length; i++){
if(input[i] < kArray[0]){
kArray[0] = input[i];
maxHeap(kArray, 0);
}
}
for (int i = kArray.length - 1; i >= 0; i--){
list.add(kArray[i]);
}
return list;
}
private void buildHeap(int[] input) {
for (int i = input.length/2 - 1; i >= 0; i--){
maxHeap(input, i);
}
}
private void maxHeap(int[] array, int i) {
int left = 2*i + 1;
int right = left + 1;
int largest = 0;
if(left < array.length && array[left] > array[i])
largest = left;
else
largest = i;
if(right < array.length && array[right] > array[largest])
largest = right;
if (largest != i){
int temp = array[i];
array[i] = array[largest];
array[largest] = temp;
maxHeap(array, largest);
}
}
}