插入排序
根据插入排序的思想，我们很容易就可以完成插入排序的代码如下。
func insertionSort(data []int) {
lo, hi := 0, len(data)
for i := lo + 1; i < hi; i++ {
for j := i; j > lo && data[j] < data[j-1]; j-- {
data[j], data[j-1] = data[j-1], data[j]
}
}
}
复制代码我们可以验证一下，确实没有问题。
package main
import (
"fmt"
)
func main() {
nums := []int{2, 3, 4, 1, 7, 9, 10, 21, 17}
insertionSort(nums)
fmt.Println(nums)
}
复制代码代码输出为，结果正确
[1 2 3 4 7 9 10 17 21]
复制代码问题
好，现在问题来了，都知道 Go 是静态语言，那么就意味着不同的数据类型可能导致上述的插入排序不可用。比如说，某天产品想要支持 uint32 的插入排序。嗯，很简单，直接 Ctrl+c + Ctrl+c 稍微修改一下。
func insertionSortUint32(data []uint32) {
lo, hi := 0, len(data)
for i := lo + 1; i < hi; i++ {
for j := i; j > lo && data[j] < data[j-1]; j-- {
data[j], data[j-1] = data[j-1], data[j]
}
}
}
复制代码谁知道哪天产品脑子又抽风，他想要支持 float32 类型的插入排序，代码可能又得加几行。
func insertionSortFloat32(data []float32) {
lo, hi := 0, len(data)
for i := lo + 1; i < hi; i++ {
for j := i; j > lo && data[j] < data[j-1]; j-- {
data[j], data[j-1] = data[j-1], data[j]
}
}
}
复制代码好像还看得下去，我们知道 Go 中的类型可不止这 3 种，再这么被搞下去是不是要爆炸了？没关系，我们有强大的 IDE 可以快速实现。😏😏😏
好了，开个玩笑。如果我们是提供一个库的形式，使用者需要一个类型，我们就得加一个类型支持，这样就没法搞事了😂
解决
首先，回到上诉的三个类型的排序中来，我们可以发现这几个排序除了数据类型是基本一致的。如果我们想用一个函数来支持所有的数据类型，我们是不是只能使用 interface来实现这个功能？但是 interface 又不支持运算操作，如果断言出来，还是跟以前一样麻烦。我们看看代码中需要对数据进行运算操作的地方。
发现排序中只有len(data)，data[j] < data[j-1]，data[j], data[j-1] = data[j-1], data[j]这三种操作 interface 不支持。如果我们让 interface 实现这三个方法不就解决了我的问题了吗？接下来我们通过这种思路修改一下我们的插入排序 。代码如下，
type Data interface {
Len() int
Less(i, j int) bool
Swap(i, j int)
}
func insertionSort(data Data) {
lo, hi := 0, data.Len()
for i := lo + 1; i < hi; i++ {
for j := i; j > lo && data.Less(j, j-1); j-- {
data.Swap(j, j-1)
}
}
}
复制代码我们使用了interface来替代写死的数据类型。如果调用方使用，只要实现 Data 接口就行了。
package main
import (
"fmt"
)
type Uint32Slice []uint32
func (u Uint32Slice) Len() int {return len(u)}
func (u Uint32Slice) Less(i, j int) bool {return u[i] < u[j]}
func (u Uint32Slice) Swap(i, j int) {u[i], u[j] = u[j], u[i]}
type Float32Slice []float32
func (u Float32Slice) Len() int {return len(u)}
func (u Float32Slice) Less(i, j int) bool {return u[i] < u[j]}
func (u Float32Slice) Swap(i, j int) {u[i], u[j] = u[j], u[i]}
func main() {
nums := Uint32Slice{2, 3, 4, 1, 7, 9, 10, 21, 17}
insertionSort(nums)
fmt.Println(nums)
float32Nums := Float32Slice{2, 3, 4, 1, 7, 9, 10, 21, 17}
insertionSort(float32Nums)
fmt.Println(float32Nums)
}
复制代码可以验证，结果没有问题。
[1 2 3 4 7 9 10 21 17]
[1 2 3 4 7 9 10 21 17]
复制代码总结
我们通过接口实现了一个支持多种数据类型的插入排序，调用者只需要实现 Data 这个接口就可以使用了，而不用去修改插入排序原有的函数定义。这样使得我们的代码抽象度更高也更灵活，当我们面临类似的需求时，接口就是答案。
