给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。 请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
时间: 2024-03-28 21:16:14 浏览: 145
python 实现给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)
5星 · 资源好评率100%
好的,这是一个程序方面的问题。我们可以使用快速选择算法来解决这个问题,时间复杂度为 O(n),空间复杂度为 O(1)。具体实现方式可以参考以下代码:
```python
def findKthLargest(nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
def partition(l, r):
pivot = nums[r]
i = l - 1
for j in range(l, r):
if nums[j] >= pivot:
i += 1
nums[i], nums[j] = nums[j], nums[i]
nums[i+1], nums[r] = nums[r], nums[i+1]
return i+1
l, r = 0, len(nums) - 1
while True:
pos = partition(l, r)
if pos == k - 1:
return nums[pos]
elif pos < k - 1:
l = pos + 1
else:
r = pos - 1
```
这个算法的思路是每次选择一个 pivot,将小于 pivot 的放在 pivot 的左边,大于 pivot 的放在 pivot 的右边,然后判断 pivot 的位置和 k 的大小关系,如果 pivot 在 k 的左边,就在右边继续寻找 pivot,否则就在左边继续寻找 pivot。这样做的时间复杂度是 O(n),由于我们每次只需要进入一个分支,所以空间复杂度是 O(1)。
阅读全文