215. 数组中的第K个最大元素(时间复杂度o(n))

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。

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

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

以空间换时间,数组下标不能为负数,故下标整体后移10000个单位,结果再返回

class Solution:
    def findKthLargest(self, nums, k):
        bucket = [0] * 20005

        for n in nums:
            bucket[n + 10000] += 1

        sum_val = 0
        for i in range(20004, -1, -1):
            sum_val += bucket[i]
            if sum_val >= k:
                return i - 10000

        return 0
  1. 创建一个长度为 20005 的数组 bucket,用于记录每个元素出现的次数。数组的索引从 -10000 到 10004,对应了可能的元素取值范围。

  2. 遍历输入数组 nums,将每个元素的出现次数记录在 bucket 数组中。具体而言,对于元素 n,在 bucket[n + 10000] 的位置加一。

  3. bucket 数组的末尾开始向前遍历,累积每个索引处的元素出现次数(从大到小累积)。当累积次数达到或超过 k 时,返回当前索引值减去 10000,即找到了第 k 大的元素。

  4. 如果遍历完整个 bucket 数组后还没有找到第 k 大的元素,返回默认值 0。

总体来说,该算法使用桶排序的思想,通过统计每个元素的出现次数,从大到小遍历桶来找到第 k 大的元素。

C语言实现

#include <stdio.h>

int findKthLargest(int* nums, int size, int k) {
    int bucket[20005] = {0};

    // Count the occurrences of each element in the bucket array.
    for (int i = 0; i < size; i++) {
        bucket[nums[i] + 10000]++;
    }

    // Traverse the bucket array from the end to find the kth largest element.
    int sum = 0;
    for (int i = 20004; i >= 0; i--) {
        sum += bucket[i];
        if (sum >= k) {
            return i - 10000;
        }
    }

    return 0;
}

int main() {
    // Example usage:
    int nums[] = {3, 1, 4, 4, 2};
    int size = sizeof(nums) / sizeof(nums[0]);
    int k = 2;

    int result = findKthLargest(nums, size, k);
    printf("The %dth largest element is: %d\n", k, result);

    return 0;
}

  • 5
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

西柚与蓝莓

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值