《剑指offer》—— 40 最小的K个数

题目

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

解法1

• 快排，一趟快排后看基准元素位置，如果恰好是k-1，则其前面的就是最小的k个数；如果小于k-1，则保留基准元素前面的，继续对后面的快排；如果大于k-1，则基准元素后面的舍弃，对前面的继续快排。
• 时间复杂度O(n)；但是会修改原数组。

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(input.length < k){
            return list;
        }
        findKMin(input,0,input.length-1,k);
        for(int i = 0; i < k; i++){
            list.add(input[i]);
        }
        return list;
    }
    
	public void findKMin(int[] a, int begin, int end,int k){
		if(begin < end){
			int pos = partition(a, begin, end);
			if(pos == k-1){
				return ;
		    }else if(pos < k-1){
		    	findKMin(a,pos+1,end,k);
		    }else{
		    	findKMin(a,begin,pos-1,k);
		    }
	    }
	}
	    
	public int partition(int[] a, int begin, int end){
	  	int pivot = a[begin];
	  	while(begin < end){
		   	while(begin < end && a[end] >= pivot)
		   		end--;
		   	a[begin] = a[end];
		   	while(begin < end && a[begin] <= pivot)
		   		begin++;
		   	a[end] = a[begin];
	 	}
	  	a[begin] = pivot;
	  	return begin;
	 }
}

解法2

• 用最大堆，用一个大小为k的容器，数组中前k个先放进去；然后从数组中读取元素，如果读入的元素小于容器中的最大值，则替换。

	public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        ArrayList<Integer> result = new ArrayList<Integer>();
        int length = input.length;
        if(k > length || k == 0){
            return result;
        }
        
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>(k, new Comparator<Integer>(){
            //重写compare，如果o2>o1，o2放前面
            @Override
            public int compare(Integer o1, Integer o2){
                return o2.compareTo(o1);
            }
        });
        
        for(int i = 0; i < length; i++){
            if(maxHeap.size() < k){
                maxHeap.offer(input[i]);
            }
            else if(maxHeap.peek() > input[i]){
                maxHeap.poll();
                maxHeap.offer(input[i]);
            }
        }
        
        for(int i = 0; i < k; i++){
            result.add(maxHeap.poll());
        }
        return result;
    }