150强访谈简单

数组中第 K 个最大元素

“数组中的第 K 个最大元素”问题的详细指南和 Python 实现。

问题陈述

简单

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

请注意,它是排序顺序中的第 k 个最大元素,而不是第 k 个不同元素。

你能以 O(n) 的复杂度解决这个问题吗?

编写一个函数 findKthLargest(nums: List[int], k: int) -> int

约束条件
  • 1 <= k <= len(nums) <= 10^5
  • -10^4 <= nums[i] <= 10^4

示例

Example 1
Input
nums = [3,2,1,5,6,4], k = 2
Output
5
Explanation

The sorted array is [1,2,3,4,5,6]. The 2nd largest element is 5.

Example 2
Input
nums = [3,2,3,1,2,4,5,5,6], k = 4
Output
4
Explanation

The sorted array is [1,2,2,3,3,4,5,5,6]. The 4th largest element is 4.

Need a Hint?
考虑使用堆/优先级队列特定的数据结构,例如集合或堆。
Edge Cases to Watch
  • 空输入结构
  • 单元素输入
  • 大数值范围

准备好解决了吗?

Open the problem in PyRun's browser-based Python editor. Your code runs fully offline — no server required.

在编辑器中打开
Found this breakdown helpful?

PyRun is built and maintained by an independent solo developer. If this helped your interview prep, consider buying a coffee!

Buy me a coffee

推荐的 Python 资源

通过相关的交互式教程、备忘单和代码比较来扩展您的知识。