46最小的k个数 47寻找第k大

46 最小的k个数

第一反应排序的话是O(nlogn),暴力是O(NK)，看了题解用优先队列，因为优先队列插入元素需要logn,我们只要让堆的大小控制在k,插入一个元素就值需要logk，这样整体的时间复杂度刚好符合要求。写优先队列很简单，只要poll offer peek,具体插入的算法不需要我们自己写，只需要用这个数据结构即可。

import java.util.*;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        if(k==0) return new ArrayList<>();
        ArrayList<Integer> res = new ArrayList<>();
        PriorityQueue<Integer> q = new PriorityQueue<>((a,b)->b-a);
        //注意这样才是最大堆，否则不写默认是最小堆
        for(int i=0;i<k;i++){
            q.offer(input[i]);
        }
        for(int i=k;i<input.length;i++){
            if(q.peek()>input[i]){//如果比最大的小，就poll，插入新元素
                q.poll();
                q.offer(input[i]);
            }
        }
        for(int i=0;i<k;i++){
            res.add(q.poll());
        }
        return res;
        
    }
}

PriorityQueue q = new PriorityQueue<>((a,b)->b-a);//最大堆

47.寻找第k大

其实用内置的快排也能符合复杂度的要求，只要

Arrays.sort(a); //默认从小到大排序
return a[n-K];

但是题目指定了快排的方法，只需要在快排的基础上稍微修改一下：

• 如果j+1==K 当前j位置的元素就刚好是第k大
• 如果j+1<K 说明比当前元素大的不足K个，返回右边区间分治
• 如果j+1>K ，返回左边区间分治

import java.util.*;

public class Solution {
    public int findKth(int[] a, int n, int K) {
        // write code here
        return Partition(a,0,n-1,K);
    }
    public int Partition(int[]a, int begin, int end, int k){
            int i=begin;
            int j=end;
            int key = a[begin];
            while(i<j){//a[begin]为基准，从左到右找到第一个小于基准的
                while(i<j&&a[j]<=key) j--;
                if(i<j) a[i++] = a[j];
                while(i<j&&a[i]>=key) i++;
                if(i<j) a[j--] = a[i];
            }
            a[j]=key;
            if(j+1==k) return a[j];
            else if(j+1<k) return Partition(a,j+1,end,k);//在右边
            else return Partition(a,begin,j-1,k);//在左边     
        
    }

}