[三分钟学算法]分治-快速排序-最小的K个数:设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。

文章目录

题目详情

题目链接

设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。

示例:

java 复制代码
输入: arr = [1,3,5,7,2,4,6,8], k = 4
输出: [1,2,3,4]

提示:

0 <= len(arr) <= 100000

0 <= k <= min(100000, len(arr))

算法原理

我们可以用快速选择算法来解决:

  1. 随机选择一个基准元素key
  2. 递归地将数组分成三部分:<key区;=key区;>key区。
  3. 分类讨论

编写代码

java 复制代码
class Solution {
    public int[] smallestK(int[] nums, int k) {
        qsort(nums, 0, nums.length - 1, k);

        int[] ret = new int[k];
        for (int i = 0; i < k; i++)
            ret[i] = nums[i];
        
        return ret;
    }

    public void qsort(int[] nums, int l, int r, int k) {
        if (l >= r) return; // 递归出口

        // 1. 随机选择一个基准元素key
        int key = nums[new Random().nextInt(r - l + 1) + l];
        int left = l - 1, right = r + 1, i = l;

        //2. 将数组分成三块
        while (i < right) {
            if (nums[i] < key) swap(nums, ++left, i++);
            else if (nums[i] == key) i++;
            else swap(nums, --right, i);
        }

        // 3.分类讨论
        int a = left - l + 1, b = right - left -1;
        if (a > k) qsort(nums, l, left, k);
        else if (a + b >= k) return;
        else qsort(nums, right, r, k - a - b);
    }

    public void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}
相关推荐
智界工具库4 分钟前
《IDEA 2025 长效使用指南:2099 年有效期配置实战之JetBrains全家桶有效》
java·ide·intellij-idea
NAGNIP4 分钟前
一文搞懂量化、剪枝和知识蒸馏都是什么?
算法
Mr.45677 分钟前
MQTT通信实现方案(Spring Boot 3 集成MQTT)
java·spring boot·后端
徐子童15 分钟前
JVM高频面试题---类加载机制
java·jvm·面试题
点云SLAM18 分钟前
GTSAM 中自定义因子(Custom Factor)的详解和实战示例
算法·机器人·slam·后端优化·gtsam·gtsam自定义因子·因子图
青草地溪水旁39 分钟前
设计模式(C++)详解——迭代器模式(2)
java·c++·设计模式·迭代器模式
9号达人41 分钟前
Java18 新特性详解与实践
java·后端·面试
我不是混子1 小时前
java浮点数精度问题及解决方案
java·后端
花心蝴蝶.1 小时前
Java 中的代理模式
java·开发语言·代理模式
萘柰奈1 小时前
LeetCode刷题记录----62.不同路径(Medium)
算法·leetcode·职场和发展