TopK的两种解法

来源：互联网发布：淘宝上卖的vr是真的吗编辑：程序博客网时间：2024/04/28 17:08

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

思路1：如果k比较大，可以直接把数组划分为2个部分，一部分为[0,k-1] ，一部分为【k，n-1】

遍历第二部分，和第一部分的最大值比较，如果更小则交换，遍历结束第一部分就是topk的结果

时间复杂度是k(n-k)

思路2：利用大顶堆，时间复杂度是nlogk

思路1：

import java.util.ArrayList;public class Solution {    public ArrayList<Integer> GetLeastNumbers_Solution(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 = this.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;    }}

思路2：

import java.util.ArrayList;public class Solution {    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {        ArrayList<Integer> list=new ArrayList<Integer>();        //检查输入的特殊情况        if(input==null || input.length<=0 || input.length<k){            return list;        }        //构建最大堆        for(int len=k/2-1; len>=0; len--){            adjustMaxHeapSort(input,len,k-1);        }        //从第k个元素开始分别与最大堆的最大值做比较，如果比最大值小，则替换并调整堆。        //最终堆里的就是最小的K个数。        int tmp;        for(int i=k; i<input.length; i++){            if(input[i]<input[0]){                tmp=input[0];                input[0]=input[i];                input[i]=tmp;                adjustMaxHeapSort(input,0,k-1);            }        }        for(int j=0; j<k; j++){            list.add(input[j]);        }        return list;    }         public void adjustMaxHeapSort(int[] input, int pos, int length){        int temp;        int child;        for(temp=input[pos]; 2*pos+1<=length; pos=child){            child=2*pos+1;            if(child<length && input[child]<input[child+1]){                child++;            }            if(input[child]>temp){                input[pos]=input[child];            }else{                break;            }        }        input[pos]=temp;    }}

0 0