最小的k个数（Java实现）

最新推荐文章于 2021-11-21 12:07:42 发布

!&?

最新推荐文章于 2021-11-21 12:07:42 发布

阅读量653

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer（Java实现）

本文链接：https://blog.youkuaiyun.com/melwx/article/details/88629440

剑指offer（Java实现）专栏收录该内容

28 篇文章

订阅专栏

博客围绕找出n个整数中最小的K个数展开。题目给出输入n个整数，需找出最小的K个数示例。解题思路是设置大小为k的容器，有空余则添加数组中的数，否则比较数组中的数和容器最大数，选最小的放入。还提及ArrayList的indexOf方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

解题思路
设置一个大小为k的容器，如果还有空余，就将数组中的数加进去，否则就比较数组中的数和容器中最大的数，选择最小的放进去。
注：ArrayList中的indexOf（int k）方法，是返回值为k的下标。

代码实现

import java.util.ArrayList;
import java.util.Collections;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> res = new ArrayList();
        if(input.length < k || k <= 0) {
            return res;
        }
        for(int i = 0; i < input.length; i++) {
            if(res.size() < k) {
                res.add(input[i]);
            }
            else {
                int max = Collections.max(res);
                if(max > input[i]) {
                    res.set(res.indexOf(max), input[i]);
                }
            }
        }
        return res;
    }
}