feat: add Swift codes for top_k article (#578)

This commit is contained in:
nuomi1
2023-07-06 03:16:14 +08:00
committed by GitHub
parent 55d8b710fa
commit b4de2c0d13
2 changed files with 37 additions and 0 deletions

View File

@@ -0,0 +1,35 @@
/**
* File: top_k.swift
* Created Time: 2023-07-02
* Author: nuomi1 (nuomi1@qq.com)
*/
import utils
/* k */
func topKHeap(nums: [Int], k: Int) -> [Int] {
// k
var heap = Array(nums.prefix(k))
// k+1 k
for i in stride(from: k, to: nums.count, by: 1) {
//
if nums[i] > heap.first! {
heap.removeFirst()
heap.insert(nums[i], at: 0)
}
}
return heap
}
@main
enum TopK {
/* Driver Code */
static func main() {
let nums = [1, 7, 6, 3, 2]
let k = 3
let res = topKHeap(nums: nums, k: k)
print("最大的 \(k) 个元素为")
PrintUtil.printHeap(queue: res)
}
}