feat: add Swift codes for counting_sort article (#438)

This commit is contained in:
nuomi1
2023-03-23 02:57:13 +08:00
committed by GitHub
parent 65e47b0748
commit 5ba85ea69d
2 changed files with 72 additions and 0 deletions
@@ -0,0 +1,70 @@
/**
* File: counting_sort.swift
* Created Time: 2023-03-22
* Author: nuomi1 (nuomi1@qq.com)
*/
/* */
//
func countingSortNaive(nums: inout [Int]) {
// 1. m
let m = nums.max()!
// 2.
// counter[num] num
var counter = Array(repeating: 0, count: m + 1)
for num in nums {
counter[num] += 1
}
// 3. counter nums
var i = 0
for num in stride(from: 0, to: m + 1, by: 1) {
for _ in stride(from: 0, to: counter[num], by: 1) {
nums[i] = num
i += 1
}
}
}
/* */
//
func countingSort(nums: inout [Int]) {
// 1. m
let m = nums.max()!
// 2.
// counter[num] num
var counter = Array(repeating: 0, count: m + 1)
for num in nums {
counter[num] += 1
}
// 3. counter
// counter[num]-1 num res
for i in stride(from: 0, to: m, by: 1) {
counter[i + 1] += counter[i]
}
// 4. nums res
// res
var res = Array(repeating: 0, count: nums.count)
for i in stride(from: nums.count - 1, through: 0, by: -1) {
let num = nums[i]
res[counter[num] - 1] = num // num
counter[num] -= 1 // 1 num
}
// 使 res nums
for i in stride(from: 0, to: nums.count, by: 1) {
nums[i] = res[i]
}
}
@main
enum CountingSort {
/* Driver Code */
static func main() {
var nums = [1, 0, 1, 2, 0, 4, 0, 2, 2, 4]
countingSortNaive(nums: &nums)
print("计数排序(无法排序对象)完成后 nums = \(nums)")
var nums1 = [1, 0, 1, 2, 0, 4, 0, 2, 2, 4]
countingSort(nums: &nums1)
print("计数排序完成后 nums1 = \(nums1)")
}
}