Golang中使用set

Go中是不提供Set類型柿赊,Set是一個集合俩功,set里的元素不能重復。

兩種思路

2 basic set implementations

使用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}
?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末术幔,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子湃密,更是在濱河造成了極大的恐慌诅挑,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,858評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件泛源,死亡現場離奇詭異拔妥,居然都是意外死亡,警方通過查閱死者的電腦和手機达箍,發(fā)現死者居然都...
    沈念sama閱讀 93,372評論 3 395
  • 文/潘曉璐 我一進店門没龙,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事硬纤〗庾遥” “怎么了?”我有些...
    開封第一講書人閱讀 165,282評論 0 356
  • 文/不壞的土叔 我叫張陵筝家,是天一觀的道長洼裤。 經常有香客問我,道長溪王,這世上最難降的妖魔是什么腮鞍? 我笑而不...
    開封第一講書人閱讀 58,842評論 1 295
  • 正文 為了忘掉前任,我火速辦了婚禮莹菱,結果婚禮上移国,老公的妹妹穿的比我還像新娘。我一直安慰自己道伟,他們只是感情好迹缀,可當我...
    茶點故事閱讀 67,857評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著皱卓,像睡著了一般裹芝。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上娜汁,一...
    開封第一講書人閱讀 51,679評論 1 305
  • 那天嫂易,我揣著相機與錄音,去河邊找鬼掐禁。 笑死怜械,一個胖子當著我的面吹牛,可吹牛的內容都是我干的傅事。 我是一名探鬼主播缕允,決...
    沈念sama閱讀 40,406評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼蹭越!你這毒婦竟也來了障本?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 39,311評論 0 276
  • 序言:老撾萬榮一對情侶失蹤响鹃,失蹤者是張志新(化名)和其女友劉穎驾霜,沒想到半個月后,有當地人在樹林里發(fā)現了一具尸體买置,經...
    沈念sama閱讀 45,767評論 1 315
  • 正文 獨居荒郊野嶺守林人離奇死亡粪糙,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,945評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現自己被綠了忿项。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片蓉冈。...
    茶點故事閱讀 40,090評論 1 350
  • 序言:一個原本活蹦亂跳的男人離奇死亡城舞,死狀恐怖,靈堂內的尸體忽然破棺而出寞酿,到底是詐尸還是另有隱情家夺,我是刑警寧澤,帶...
    沈念sama閱讀 35,785評論 5 346
  • 正文 年R本政府宣布熟嫩,位于F島的核電站秦踪,受9級特大地震影響,放射性物質發(fā)生泄漏掸茅。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,420評論 3 331
  • 文/蒙蒙 一柠逞、第九天 我趴在偏房一處隱蔽的房頂上張望昧狮。 院中可真熱鬧,春花似錦板壮、人聲如沸逗鸣。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,988評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽撒璧。三九已至,卻和暖如春笨使,著一層夾襖步出監(jiān)牢的瞬間卿樱,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,101評論 1 271
  • 我被黑心中介騙來泰國打工硫椰, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留繁调,地道東北人。 一個月前我還...
    沈念sama閱讀 48,298評論 3 372
  • 正文 我出身青樓靶草,卻偏偏與公主長得像蹄胰,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子奕翔,可洞房花燭夜當晚...
    茶點故事閱讀 45,033評論 2 355

推薦閱讀更多精彩內容