build

chapter_sorting/radix_sort.md

@@ -634,6 +634,56 @@ $$
     }
     ```
 
+=== "Rust"
+
+    ```rust title="radix_sort.rs"
+    /* 获取元素 num 的第 k 位，其中 exp = 10^(k-1) */
+    fn digit(num: i32, exp: i32) -> usize {
+        // 传入 exp 而非 k 可以避免在此重复执行昂贵的次方计算
+        return ((num / exp) % 10) as usize;
+    }
+
+    /* 计数排序（根据 nums 第 k 位排序） */
+    fn counting_sort_digit(nums: &mut [i32], exp: i32) {
+        // 十进制的位范围为 0~9 ，因此需要长度为 10 的桶
+        let mut counter = [0; 10];
+        let n = nums.len();
+        // 统计 0~9 各数字的出现次数
+        for i in 0..n {
+            let d = digit(nums[i], exp); // 获取 nums[i] 第 k 位，记为 d
+            counter[d] += 1; // 统计数字 d 的出现次数
+        }
+        // 求前缀和，将“出现个数”转换为“数组索引”
+        for i in 1..10 {
+            counter[i] += counter[i - 1];
+        }
+        // 倒序遍历，根据桶内统计结果，将各元素填入 res
+        let mut res = vec![0; n];
+        for i in (0..n).rev() {
+            let d = digit(nums[i], exp);
+            let j = counter[d] - 1; // 获取 d 在数组中的索引 j
+            res[j] = nums[i]; // 将当前元素填入索引 j
+            counter[d] -= 1; // 将 d 的数量减 1
+        }
+        // 使用结果覆盖原数组 nums
+        for i in 0..n {
+            nums[i] = res[i];
+        }
+    }
+
+    /* 基数排序 */
+    fn radix_sort(nums: &mut [i32]) {
+        // 获取数组的最大元素，用于判断最大位数
+        let m = *nums.into_iter().max().unwrap();
+        // 按照从低位到高位的顺序遍历
+        let mut exp = 1;
+        while exp <= m {
+            counting_sort_digit(nums, exp);
+            exp *= 10;
+        }
+    }
+    ```
+
 !!! question "为什么从最低位开始排序？"
 
     在连续的排序轮次中，后一轮排序会覆盖前一轮排序的结果。举例来说，如果第一轮排序结果 $a < b$ ，而第二轮排序结果 $a > b$ ，那么第二轮的结果将取代第一轮的结果。由于数字的高位优先级高于低位，我们应该先排序低位再排序高位。