从两个数组/切片中获取交集和排除项的最有效方法是什么?

时间:2018-08-31 17:46:42

标签: arrays go slice

给出两个数组或切片,例如:

a := []int{1, 2, 3, 4, 5}
b := []int{3, 4, 5, 6, 7, 8, 9}

切片可能没有排序,顺序也没有关系。

计算值的最有效方法是什么,这样您最终得到两个切片的公共元素,而一个元素中却存在另一个而不是另一个元素的其余元素,即上面给出的两个数组的返回值将是:

common := []int{3, 4, 5}
inAButNotB := []int{1, 2}
inBButNotA := []int{6, 7, 8, 9}

计算相交点很容易,将一个切片转换为地图,然后在一个切片上进行迭代以查看值是否存在。有没有一种方法可以计算同一循环中的其他两个值?

1 个答案:

答案 0 :(得分:4)

O(len(a) + len(b))是有效的。例如,

package main

import (
    "fmt"
)

func main() {
    a := []int{1, 2, 3, 4, 5}
    b := []int{3, 4, 5, 6, 7, 8, 9}
    fmt.Println(a)
    fmt.Println(b)

    m := make(map[int]uint8)
    for _, k := range a {
        m[k] |= (1 << 0)
    }
    for _, k := range b {
        m[k] |= (1 << 1)
    }

    var inAAndB, inAButNotB, inBButNotA []int
    for k, v := range m {
        a := v&(1<<0) != 0
        b := v&(1<<1) != 0
        switch {
        case a && b:
            inAAndB = append(inAAndB, k)
        case a && !b:
            inAButNotB = append(inAButNotB, k)
        case !a && b:
            inBButNotA = append(inBButNotA, k)
        }
    }
    fmt.Println(inAAndB)
    fmt.Println(inAButNotB)
    fmt.Println(inBButNotA)
}

游乐场:https://play.golang.org/p/RvGaC9Wfjiv

输出:

[1 2 3 4 5]
[3 4 5 6 7 8 9]
[3 4 5]
[1 2]
[8 6 7 9]

  

The Go Programming Language Specification

&    bitwise AND            integers
|    bitwise OR             integers
^    bitwise XOR            integers
&^   bit clear (AND NOT)    integers

<<   left shift             integer << unsigned integer
>>   right shift            integer >> unsigned integer

uint8有8位。位0(1 << 0,左移1)是a,位1(1 << 1;左移1)是b。对于uint8位,00000001a00000010b00000011ab00000000既不是a也不是b|运算符设置一个位,&运算符读取一个位。


  

The Go Programming Language Specification

     

Map types

     

地图是一种无序的一组类型的元素,称为   元素类型,由另一种类型的唯一键集索引,称为   密钥类型。

     

必须为操作数完全定义比较运算符==和!=   密钥类型;因此,键类型不能为函数,映射或   切片。如果键类型是接口类型,则这些比较   必须为动态键值定义运算符;失败会   导致运行时出现紧急情况。

该算法适用于其元素可以是映射键的任何切片类型。必须为键类型的操作数完全定义比较运算符==和!=。