剑指offer 40.最小的k个数。双百0ms

Split_token

已于 2022-01-20 10:58:21 修改

阅读量308

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签：排序算法算法 leetcode

于 2022-01-20 10:58:10 首次发布

本文链接：https://blog.youkuaiyun.com/Split_token/article/details/122597056

LeetCode 专栏收录该内容

221 篇文章

订阅专栏

本文介绍了一种简单高效的计数排序算法实现方法。通过使用计数数组来统计每个元素出现的次数，进而得到前k个最小的整数。此算法适用于整数范围已知且不大的情况。

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

计数排序

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if(k == 0 || arr.length == 0) return new int[0];
        int[] count = new int[10001];
        for(int num : arr) count[num]++;
        
        int[] result = new int[k];
        int index = 0;
        for(int i = 0; i < count.length; i++){
            while(count[i]-- > 0 && index < k) result[index++] = i;
            if(index == k) break;
        }
        return result;
    }
}