剑指offer面试题30：最小的k个数

阅读目录

一、题目描述
二、解题思路

回到顶部

一、题目描述

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

回到顶部

二、解题思路

　　1、思路1

　　首先对数组进行排序，然后取出前k个数。排序方法可选用快速排序，时间复杂度为O(nl*ogn)。示例代码：

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.length<k||k==0)
            return list;
        Arrays.sort(input);   
        for(int i=0;i<k;i++)
            list.add(input[i]);
        return list;
    }
}

　　2、思路2

　　首先将这堆数据中前k个数装入list，并找到这个k个数的最大数max(k)，然后从第k+1个数开始向后找，如果有小于这个max(k)的，则替换掉这个数，然后重新这k个数中找出最大的max(k)。这样一直向后扫描。时间复杂度为O(k*n)。示例代码：

public class Solution_getLeastNumbers
{
    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;
    }
}

　　3、思路3

　　扫描数组，每遍找出最小的那个数，复杂度为O(k*n)。示例代码：

public ArrayList<Integer> GetLeastNumbers_Solution3(int[] input, int k)
    {
        ArrayList<Integer> list = new ArrayList<Integer>();

        if (input.length < k || k == 0)
            return list;

        for (int i = 0; i < input.length; i++)
            list.add(input[i]);
        
        ArrayList<Integer> result=new ArrayList<>();
        int num=0;
        while(num<k)
        {
            int minNum=list.get(0);
            for(int i=0;i<list.size();i++)
            {
                if(list.get(i)<minNum)
                {
                    minNum=list.get(i);
                }
            }
            list.remove((Object)minNum);
            result.add(minNum);
            num++;
        }
        return result;
    }