Golang 使用Map實現去重與set的功能操作

Posted on by

假設現在有一個需求

將以下 url 保存起來，要求不重復

www.baidu.com
www.sina.com.cn
www.hao123.com
www.baidu.com
www.hao123.com

可以利用 map 的 key 唯一性的問題處理

package main 
var set = map[string]bool { 
}
 
func main() {
    ...
    url := xxx
    if set[url] {
        // 表示集合中已經存在
        return
    }
 
    set[url] = true           // 否則如果不存在，設置為true
}

// 完成後，set的所有的key值為不重復的值

在此之前，需要確定 golang 支持的 map 的 key 數量

可以手寫暴力寫入測試

package main 
import (
 "fmt"
)
 
var test = map[int]int { 
}
 
func main() {
 for i := 0; i < 10000000; i++ {        // 測試對1000萬key的支持，完美支持
  fmt.Println(i)
  test[i] = i
 }
}

補充：golang map/set類型

map類型

1. 基礎特性

map是一種無序的鍵值對的集合; 所以可以類似數組/slice一樣進行迭代

map的值可以使內建類型, 也可以是struct類型

內部使用hash表實現, map的hash表包含瞭一個collection of buckets（桶集合）

2. 聲明與初始化

package main
import (
    "fmt"
)
// map[keyType]valueType
func initMap() {
    // 通過make方法創建
    dict := make(map[string]int)
    dict["age"] = 18
    // 直接創建
    dict2 := map[string]string{"name":"xiaoming", "phone":"135xxx"}
    dict2["addr"] = "Guangzhou"
    fmt.Printf("%v\n", dict2)
}
func main() {
    initMap()
}

3. 元素訪問

package main
import (
    "fmt"
)
type Student struct {
    name string
    grade int
}
func useMap() {
    //使用前應該先初始化, 否則panic報錯
    // var map1 map[string]string
    // map1["a"] = "b" // will panic
    map1 := make(map[string]Student)
    map1["s1"] = Student{name:"xiaomo", grade:1}
    fmt.Printf("%v\n", map1)
}
func main() {
    useMap()
}

4. 在函數中傳遞map

在函數間傳遞map對象, 是傳遞引用而不是拷貝; 因此在函數中對map進行瞭修改, 引用到它的地方也會相應修改

package main
import (
    "fmt"
)
type Student struct {
    name string
    grade int
}
func useMap() {
    map1 := make(map[string]Student)
    map1["s1"] = Student{name:"xiaomo", grade:1}
    // 作為函數參數傳遞
    printMap(map1)
}
func printMap(m map[string]Student) {
    fmt.Printf("currentMap: %v\n", m)
}
func main() {
    useMap()
}

Set類型

golang沒有內置Set類型, 可以自定義實現。

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：