just go! 關注golang, 希望找到同樣喜歡golang的朋友.

在go中使用"泛型"

[2012-05-21 翻譯自這裡, 對原文有所擴充套件, 也有所刪減.] 在進入泛型的話題之前, 首先實現對int slice(可以看做int陣列)的氣泡排序:

 1 func BubbleSort(array []int) {
 2     for i := 0; i < len(array); i++ {
 3         for j := 0; j < len(array)-i-1; j++ {
 4             if array[j] > array[j+1] {
 5                 // 交換
 6                 array[j], array[j+1] = array[j+1], array[j]
 7             }
 8         }
 9     }
10 } 

如你所見, 上面的程式碼僅適用於對int陣列進行排序, 如果想要對string陣列排序, 不得不另寫一個. 是否可以只寫一個通用的泛型程式, 以便對所有型別的陣列(甚至是任意資料)進行氣泡排序? 很遺憾, go不支援java中的標記式泛型. 但是我們可以使用go的interface實現類似的功能. interface用於定義方法的集合. 在氣泡排序中, 我們可以將排序操作分解成3個方法: Len()方法負責計算長度, Less(i, j)方法負責比較大小, Swap(i, j)方法負責進行交換.

1 // 定義可排序介面
2 type Sortable interface {
3     Len() int
4     Less(int, int) bool
5     Swap(int, int)
6 }

可以說, 任何實現了Sortable介面的資料型別都可進行氣泡排序, 下面是對Bubblesort方法的改造:

 1 func BubbleSortable(arr Sortable) {
 2     length := arr.Len()
 3     for i := 0; i < length; i++ {
 4         for j := i; j < length-i-1; j++ {
 5             if arr.Less(j, j+1) {
 6                 arr.Swap(j, j+1)
 7             }
 8         }
 9     }
10 } 

BubbleSortable函式可以對所有Sortable型別的資料進行排序. 那麼, 哪些資料是Sortable型別的呢? 很簡單, 那些實現了Len(), Less(i, j), Swap(i, j)方法的資料型別都是Sortable的. 如果想要對int陣列進行氣泡排序, 只需讓int陣列實現以上3個方法即可:

 1 // 定義IntArr型別
 2 type IntArr []int
 3 // 給IntArr提供Len方法
 4 func (arr IntArr) Len() int {
 5     return len(arr)
 6 }
 7 // 給IntArr提供Less方法
 8 func (arr IntArr) Less(i int, j int) bool {
 9     return arr[i] < arr[j]
10 }
11 // 給IntArr提供Swap方法
12 func (arr IntArr) Swap(i int, j int) {
13     arr[i], arr[j] = arr[j], arr[i]
14 } 

至此, 我們定義了新的型別: IntArr(實際上就是int陣列), 併為IntArr增加了Len, Less, Swap方法. 注意, 在go中, 實現介面不需要像java那樣顯式的說明對某個介面的implements, 只需要為型別提供所有interface中定義的方法即可. 此例中, 我們給IntArr提供了所有Sortable中定義的方法, 所以IntArr已經實現了Sortable介面. 接下來要做的是將IntArr型別的資料傳遞給BubbleSortable函式就可以了:

1 intarr := IntArr{2, 3, 1, -9, 0}
2 // 呼叫排序方法
3 BubbleSortable(intarr)
4 // 輸出排序之後的資料
5 fmt.Printf("sorted int arr is: %v\n", intarr)

同樣的, 如果想對string陣列進行氣泡排序, 也只需要讓string陣列實現Sortable介面中定義的所有方法:

 1 type StrArr []string
 2 func (arr StrArr) Len() int {
 3     return len(arr)
 4 }
 5 func (arr StrArr) Less(i int, j int) bool {
 6     return arr[i] < arr[j]
 7 }
 8 func (arr StrArr) Swap(i int, j int) {
 9     arr[i], arr[j] = arr[j], arr[i]
10 } 

測試程式碼如下:

1 strarr := StrArr{"nut", "ape", "elephant", "zoo", "go"}
2 Bubblesort(strarr)
3 fmt.Printf("sorted string arr is: %v\n", strarr)

現在, 你可以對任意資料進行氣泡排序了, 只需要該型別實現了Len, Less, Swap方法.