LeetCode - 数组中的第K个最大元素（Swift实现）

要求:

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

使用快排进行解决。

class Solution {
    func findKthLargest(_ nums: [Int], _ k: Int) -> Int {
        
        let quickSortResult = quickSort(nums, 0, nums.count - 1)
        let resultNums = quickSortResult.0
        let keyIndex = quickSortResult.1
        
        if keyIndex + 1 == k {
            
        }else if keyIndex + 1 > k {
            return findKthLargest(Array(resultNums[0..<keyIndex]), k)
        }else {
            let index = resultNums.index(after: keyIndex)
            let endIndex = resultNums.endIndex
            return findKthLargest(Array(resultNums[index..<endIndex]), k - keyIndex - 1)
        }
        return resultNums[keyIndex]
    }
    
    private
    func quickSort(_ nums: [Int], _ left: Int, _ right: Int) -> ([Int], Int) {
        var numsArray = nums
        var i = left
        var j = right
        let key = numsArray[i]
        
        while i != j {
            while j > i && numsArray[j] <= key {
                j -= 1
            }
            numsArray.swapAt(i, j)
            
            while j > i && numsArray[i] >= key {
                i += 1
            }
            numsArray.swapAt(i, j)
        }
        numsArray[i] = key
        return (numsArray, i)
    }
}