(数组)6. 无序数组中找出第K大的元素

在未排序的数组中寻找第 k 大的元素,通过切分操作缩小搜索范围,例如在示例中当 k=2 时,输出为 5;当 k=4 时,输出为 4。

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

在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4

题解思路

每次切分操作总能排定一个元素,还能够知道这个元素它最终所在的位置,这样每经过一次 partition(切分)操作就能缩小搜索的范围

实现代码

public class Solution {
    public int findKthLargest(int[] nums, int k) {
        int len = nums.length;
        int left = 0;
        int right = len - 1;
        // 转换一下,第 k 大元素的索引是 len - k
        int target = len - k;
        while (true) {
            int index = useSort(nums, left, right);
            if (index == target) {
                return nums[index];
            } else if (index < target) {
                left = index + 1;
            } else if (index > target) {
                right = index - 1;
            }
        }
    }

	/**
     * 在数组 nums 的子区间 [left, right] 执行 partition 操作,返回 nums[left] 排序以后应该在的位置
     * 在遍历过程中保持循环不变量的语义
     * 1、[left + 1, i] < nums[left]
     * 2、(i, j] >= nums[left]
     */
    private int useSort(int[] nums, int left, int right) {
        int base = nums[left];
        int i = left;
        for (int j = left + 1; j <= right; j++) {
        // 小于 pivot 的元素都被交换到前面
            if (nums[j] < base) {
                i++;
                swap(nums, i, j);
            }
        }
        swap(nums, i, left);
        return i;
    }

    private void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}

leetcode题目链接

215. 数组中的第K个最大元素

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值