Go中是不提供Set類型柿赊,Set是一個集合俩功,set里的元素不能重復。
兩種思路
使用map實現
在Golang中通常使用map來實現set碰声,map中的key為唯一值诡蜓,這與set的特性一致。
簡單實現胰挑,如下:
set := make(map[string]bool) // New empty set
set["Foo"] = true // Add
for k := range set { // Loop
fmt.Println(k)
}
delete(set, "Foo") // Delete
size := len(set) // Size
exists := set["Foo"] // Membership
map的value值是布爾型蔓罚,這會導致set多占用內存空間,解決這個問題瞻颂,則可以將其替換為空結構豺谈。在Go中,空結構通常不使用任何內存贡这。
unsafe.Sizeof(struct{}{}) // 結果為 0
優(yōu)化后茬末,如下:
type void struct{}
var member void
set := make(map[string]void) // New empty set
set["Foo"] = member // Add
for k := range set { // Loop
fmt.Println(k)
}
delete(set, "Foo") // Delete
size := len(set) // Size
_, exists := set["Foo"] // Membership
golang-set
golang-set-A simple set type for the Go language. Also used by Docker, 1Password, Ethereum.
在github上已經有了一個成熟的包,名為golang-set盖矫,包中提供了線程安全和非線程安全的set丽惭。提供了五個set函數:
// NewSet創(chuàng)建并返回空集的引用,結果集上的操作是線程安全的
func NewSet(s ...interface{}) Set {}
// NewSetFromSlice從現有切片創(chuàng)建并返回集合的引用辈双,結果集上的操作是線程安全的
func NewSetFromSlice(s []interface{}) Set {}
// NewSetWith創(chuàng)建并返回具有給定元素的新集合责掏,結果集上的操作是線程安全的
func NewSetWith(elts ...interface{}) Set {}
// NewThreadUnsafeSet創(chuàng)建并返回對空集的引用,結果集上的操作是非線程安全的
func NewThreadUnsafeSet() Set {}
// NewThreadUnsafeSetFromSlice創(chuàng)建并返回對現有切片中集合的引用湃望,結果集上的操作是非線程安全的换衬。
func NewThreadUnsafeSetFromSlice(s []interface{}) Set {}
簡單案例,如下:
package main
import (
"fmt"
"github.com/deckarep/golang-set"
)
func main() {
// 默認創(chuàng)建的線程安全的证芭,如果無需線程安全
// 可以使用 NewThreadUnsafeSet 創(chuàng)建瞳浦,使用方法都是一樣的。
s1 := mapset.NewSet(1, 2, 3, 4)
fmt.Println("s1 contains 3: ", s1.Contains(3))
fmt.Println("s1 contains 5: ", s1.Contains(5))
// interface 參數檩帐,可以傳遞任意類型
s1.Add("poloxue")
fmt.Println("s1 contains poloxue: ", s1.Contains("poloxue"))
s1.Remove(3)
fmt.Println("s1 contains 3: ", s1.Contains(3))
s2 := mapset.NewSet(1, 3, 4, 5)
// 并集
fmt.Println(s1.Union(s2))
}
結果為:
s1 contains 3: true
s1 contains 5: false
s1 contains poloxue: true
s1 contains 3: false
Set{1, 2, 4, poloxue, 3, 5}