下面由golang教學專欄跟大家介紹用go語言實作找出兩個陣列的異同,希望對需要的朋友有幫助!
最近專案上碰到個小需求,輸入是兩個數組,一個舊數組一個新數組,要求取得新數組相對舊數組所有新增和刪除的元素,例如:
输入: arr_old: {"1", "2", "4", "5", "7", "9"} arr_new: {"2", "3", "4", "6", "7"} 返回: arr_added: {"3", "6"} arr_deleted: {"1", "5", "9"}
go的標準庫中沒有類似的直接比較的方法,需要自己具體實現,最簡單的方法當然是舊數組的每個元素去新數組,找不到的就是刪除的,然後新數組的元素再挨個去舊數組找一遍,找不到就是新增的,但這個方法效率實在太低了。
這裡我使用了一種基於集合運算的思想,即分別求兩個數組的交集和並集,並集減去交集就是所有發生變化的元素(要么是新增的,要嘛是刪除的),遍歷這個集合中的元素去舊數組中查找,如果在舊數組中找到,那麼就是刪除掉的元素;反之,如果找不到,則一定能在新數組中找到(用不著真的再去遍歷一次),那麼就是新增的元素。
上程式碼,這裡有個技巧,就是利用go中map鍵唯一性的特性,用陣列的元素作為map的key,透過map實現快速查找。
package main import ( "fmt" ) func main() { //fmt.Println("Hello World!") src := []string{"1", "2", "4", "5", "7", "9"} dest := []string{"2", "3", "4", "6", "7"} added, removed := Arrcmp(src, dest) fmt.Printf("add: %v\nrem: %v\n", added, removed) } func Arrcmp(src []string, dest []string) ([]string, []string) { msrc := make(map[string]byte) //按源数组建索引 mall := make(map[string]byte) //源+目所有元素建索引 var set []string //交集 //1.源数组建立map for _, v := range src { msrc[v] = 0 mall[v] = 0 } //2.目数组中,存不进去,即重复元素,所有存不进去的集合就是并集 for _, v := range dest { l := len(mall) mall[v] = 1 if l != len(mall) { //长度变化,即可以存 l = len(mall) } else { //存不了,进并集 set = append(set, v) } } //3.遍历交集,在并集中找,找到就从并集中删,删完后就是补集(即并-交=所有变化的元素) for _, v := range set { delete(mall, v) } //4.此时,mall是补集,所有元素去源中找,找到就是删除的,找不到的必定能在目数组中找到,即新加的 var added, deleted []string for v, _ := range mall { _, exist := msrc[v] if exist { deleted = append(deleted, v) } else { added = append(added, v) } } return added, deleted }
運行結果:
add: [6 3] rem: [1 5 9]
歡迎大家交流效率更高的方法。
#以上是詳解用go語言實作找出兩個陣列的異同的詳細內容。更多資訊請關注PHP中文網其他相關文章!