Merge pull request #274 from Reanon/feat-go-sort

Add radix_sort to go codes

codes/go/chapter_sorting/radix_sort.go

@@ -0,0 +1,61 @@
+// File: radix_sort.go
+// Created Time: 2023-01-18
+// Author: Reanon (793584285@qq.com)
+
+package chapter_sorting
+
+import "math"
+
+/* 获取元素 num 的第 k 位，其中 exp = 10^(k-1) */
+func digit(num, exp int) int {
+	// 传入 exp 而非 k 可以避免在此重复执行昂贵的次方计算
+	return (num / exp) % 10
+}
+
+/* 计数排序（根据 nums 第 k 位排序） */
+func countSort(nums []int, exp int) {
+	// 十进制的各位数字范围为 0~9 ，因此需要长度为 10 的桶
+	bucket := make([]int, 10)
+	n := len(nums)
+	// 借助桶来统计 0~9 各数字的出现次数
+	for i := 0; i < n; i++ {
+		d := digit(nums[i], exp) // 获取 nums[i] 第 k 位，记为 d
+		bucket[d]++              // 统计数字 d 的出现次数
+	}
+	// 求前缀和，将“出现个数”转换为“数组索引”
+	for i := 1; i < 10; i++ {
+		bucket[i] += bucket[i-1]
+	}
+	// 倒序遍历，根据桶内统计结果，将各元素填入暂存数组 tmp
+	tmp := make([]int, n)
+	for i := n - 1; i >= 0; i-- {
+		d := digit(nums[i], exp)
+		j := bucket[d] - 1 // 获取 d 在数组中的索引 j
+		tmp[j] = nums[i]   // 将当前元素填入索引 j
+		bucket[d]--        // 将 d 的数量减 1
+	}
+	// 将 tmp 复制到 nums
+	for i := 0; i < n; i++ {
+		nums[i] = tmp[i]
+	}
+}
+
+/* 基数排序 */
+func radixSort(nums []int) {
+	// 获取数组的最大元素，用于判断最大位数
+	max := math.MinInt
+	for _, num := range nums {
+		if num > max {
+			max = num
+		}
+	}
+	// 按照从低位到高位的顺序遍历
+	for exp := 1; max >= exp; exp *= 10 {
+		// 对数组元素的第 k 位执行「计数排序」
+		// k = 1 -> exp = 1
+		// k = 2 -> exp = 10
+		// k = 3 -> exp = 100
+		// 即 exp = 10^(k-1)
+		countSort(nums, exp)
+	}
+}

codes/go/chapter_sorting/radix_sort_test.go

@@ -0,0 +1,17 @@
+// File: radix_sort_test.go
+// Created Time: 2023-01-18
+// Author: Reanon (793584285@qq.com)
+
+package chapter_sorting
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestRadixSort(t *testing.T) {
+	/* 基数排序 */
+	nums := []int{23, 12, 3, 4, 788, 192}
+	radixSort(nums)
+	fmt.Println("基数排序完成后 nums = ", nums)
+}