最近项目上碰到个小需求,输入是两个数组,一个旧数组一个新数组,要求获取新数组相对旧数组所有新增和删除的元素,例如:

    1. 输入:
    2. arr_old: {"1", "2", "4", "5", "7", "9"}
    3. arr_new: {"2", "3", "4", "6", "7"}
    4. 返回:
    5. arr_added: {"3", "6"}
    6. arr_deleted: {"1", "5", "9"}

    go的标准库中没有类似的直接比较的方法,需要自己具体实现,最简单的方法当然是旧数组的每个元素去新数组,找不到的就是删除的,然后新数组的元素再挨个去旧数组找一遍,找不到就是新增的,但这个方法效率实在太低了。
    这里我使用了一种基于集合运算的思想,即分别求两个数组的交集和并集,并集减去交集就是所有发生变化的元素(要么是新增的,要么是删除的),遍历这个集合中的元素去旧数组中查找,如果在旧数组中找到,那么就是删除掉的元素;反之,如果找不到,则一定能在新数组中找到(用不着真的再去遍历一次),那么就是新增的元素。
    上代码,这里有个技巧,就是利用go中map键唯一性的特性,用数组的元素作为map的key,通过map来实现快速查找。

    1. package main
    2. import (
    3. "fmt"
    4. )
    5. func main() {
    6. //fmt.Println("Hello World!")
    7. src := []string{"1", "2", "4", "5", "7", "9"}
    8. dest := []string{"2", "3", "4", "6", "7", "99", "992"}
    9. added, removed := arrayDiff(src, dest)
    10. fmt.Printf("add: %v\nrem: %v\n", added, removed)
    11. }
    12. func arrayDiff(src []string, dest []string) ([]string, []string) {
    13. msrc := make(map[string]byte) //按源数组建索引
    14. mall := make(map[string]byte) //源+目所有元素建索引
    15. var set []string //交集
    16. //1.源数组建立map
    17. for _, v := range src {
    18. msrc[v] = 0
    19. mall[v] = 0
    20. }
    21. //2.目数组中,存不进去,即重复元素,所有存不进去的集合就是并集
    22. for _, v := range dest {
    23. l := len(mall)
    24. mall[v] = 1
    25. if l != len(mall) { //长度变化,即可以存
    26. l = len(mall)
    27. } else { //存不了,进并集
    28. set = append(set, v)
    29. }
    30. }
    31. //3.遍历交集,在并集中找,找到就从并集中删,删完后就是补集(即并-交=所有变化的元素)
    32. for _, v := range set {
    33. delete(mall, v)
    34. }
    35. //4.此时,mall是补集,所有元素去源中找,找到就是删除的,找不到的必定能在目数组中找到,即新加的
    36. var added, deleted []string
    37. for v, _ := range mall {
    38. _, exist := msrc[v]
    39. if exist {
    40. deleted = append(deleted, v)
    41. } else {
    42. added = append(added, v)
    43. }
    44. }
    45. return added, deleted
    46. }