【数组】No. 0215 数组中第 k 个最大的元素【中等】👉力扣对应题目指路
希望对你有帮助呀!!💜💜 如有更好理解的思路,欢迎大家留言补充 ~ 一起加油叭 💦
欢迎关注、订阅专栏 【力扣详解】谢谢你的支持!
⭐ 题目描述:给定整数数组 nums 和整数 k,请返回数组中第 k 大的元素,要求时间复杂度为 O(n)
- 示例:
输入: [3,2,1,5,6,4], k = 2
输出: 5
🔥 思路:第 k 大说明只用摆对第 k 个元素的有序位置 (左边均大于等于,右边均小于等于即可)
- 根据上述思路,首先想到快速排序
- 另外,由于并不需要把数组元素全部有序排列,所以可裁剪快排以
加速
- 可对比原始快排和本题代码下标注
# 加速
的部分
⭐题目准备之快速排序:一定要先掌握快速排序 ❗❗❗ 放在最后面啦,供不熟悉的友友参考~
参考如上思路,给出详细步骤如下:
- 步骤一⭐写出快速排序基本代码
- 放在最后面啦,供不熟悉的友友参考~
- 步骤二⭐加速:优化代码,仅以摆对第 k 个元素为目标,跳过对不必要的区间进行排序
- 情况一:当前中值 == 目标位置:排序完成,返回
- 情况二:当前中值 < 目标位置:继续递归排序右区间 (跳过左区间,因为其内的值均 < 当前中值)
- 情况三:当前中值 > 目标位置:继续递归排序左区间 (跳过右区间,因为其内的值均 > 当前中值)
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
def traversal(L, H): # ----------------------------- step 1
if L >= H: return
low = L
high = H
val = nums[L]
while low < high:
while low < high:
if nums[high] < val:
high -= 1
continue
else:
nums[low] = nums[high]
low += 1
break
while low < high:
if nums[low] > val:
low += 1
continue
else:
nums[high] = nums[low]
high -= 1
break
nums[low] = val
# -------------------------------- 加速! # ----step 2
if low == k-1: return # -------------------- step 2.1
if low < k-1: traversal(low+1, H) # ------- step 2.2
if low > k-1: traversal(L, low-1) # ------- step 2.3
traversal(0, len(nums)-1)
return nums[k-1]
⭐⭐⭐ 题目准备之快速排序:一定要先掌握快速排序 ❗❗❗
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
def traversal(L, H):
if L >= H: ## 当前处理的区间长度为 0,返回
return
low = L
high = H
val = nums[L]
while low < high:
while low < high: ## 从 high 向左找(比 mid_val: val)更大的值并交换
if nums[high] < val:
high -= 1
continue
else: ## 找到,交换
nums[low] = nums[high]
low += 1
break
while low < high: ## 从 low 向右找(比 mid_val: val)更小的值并交换
if nums[low] > val:
low += 1
continue
else: ## 找到,交换
nums[high] = nums[low]
high -= 1
break
# -------------------------------- 未加速
nums[low] = val
traversal(L, low-1)
traversal(low+1, H)
traversal(0, len(nums)-1)
return nums[k-1]
希望对你有帮助呀!!💜💜 如有更好理解的思路,欢迎大家留言补充 ~ 一起加油叭 💦
欢迎关注、订阅专栏 【力扣详解】谢谢你的支持!
🔥 LeetCode 热题 HOT 100