登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-03-31 09:39:33 418浏览 收藏

本篇文章给大家分享《从两个数组/切片中获取交集和排除的最有效方法是什么?》,覆盖了Golang的常见基础知识,其实一个语言的全部知识点一篇文章是不可能说完的,但希望通过这些问题,让读者对自己的掌握程度有一定的认识(B 数),从而弥补自己的不足,更好的掌握它。

问题内容

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

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}

很容易计算交集,将一个切片转换为地图,然后迭代该切片以查看值是否存在。有没有办法在同一循环中计算其他两个值?


解决方案


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 左移 0)是 a,位 1(1 << 1;1 左移 1)是 b。对于 uint8 位,00000001a00000010b00000011azqbendczq b 和 b,并且 00000000 既不是 a 也不是 b| 运算符设置一位,& 运算符读取一位。

The Go Programming Language Specification

Map types

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

必须为操作数完全定义比较运算符 == 和 != 钥匙类型;因此键类型不能是函数、映射或 片。如果键类型是接口类型,则这些比较 必须为动态键值定义运算符;失败将 导致运行时恐慌。

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

今天关于《从两个数组/切片中获取交集和排除的最有效方法是什么?》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>