LeetCode - #215 数组中的第 K 个最大元素（Top 100）

持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第19天，点击查看活动详情

前言

本题为 LeetCode 前 100 高频题

我们社区陆续会将顾毅（Netflix 增长黑客，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新到 213 期，我们会保持更新时间和进度（周一、周三、周五早上 9:00 发布），每期的内容不多，我们希望大家可以在上班路上阅读，长久积累会有很大提升。

不积跬步，无以至千里；不积小流，无以成江海，Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言，我们会尽力满足大家的需求。

难度水平：中等

1. 描述

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

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

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

2. 示例

示例 1

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

示例 2

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

提示：

• 1 <= k <= nums.length <= 10^5
• -10^4 <= nums[i] <= 10^4

3. 答案

class KthLargestElementInArray {
    func findKthLargest(_ nums: [Int], _ k: Int) -> Int {
        guard let pivot = nums.first else {
            fatalError("Invalid Input")
        }
       
        let leftPart = nums.filter { $0 > pivot }
        let middlePart = nums.filter { $0 == pivot}
        let rightPart = nums.filter { $0 < pivot }
        
        if nums.count == middlePart.count {
            return pivot
        }
        
        if leftPart.count > k - 1 {
            return findKthLargest(leftPart, k)
        } else if k - leftPart.count <= middlePart.count {
            return findKthLargest(middlePart, k - leftPart.count)
        } else {
            return findKthLargest(rightPart, k - leftPart.count - middlePart.count)
        }
    }
}

• 主要思想：快速排序。
• 时间复杂度：O(nlogn)
• 空间复杂度：O(n)

该算法题解的仓库：LeetCode-Swift

点击前往 LeetCode 练习

关于我们

我们是由 Swift 爱好者共同维护，我们会分享以 Swift 实战、SwiftUI、Swift 基础为核心的技术内容，也整理收集优秀的学习资料。