golang中判断两个slice是否相等

程序员文章站 2022-06-17 08:54:20

在golang中我们可以轻松地通过来判断两个数组（array）是否相等，但遗憾的是slice并没有相关的运算符，当需要判断两个slice是否相等时我们只能另寻捷径了。 slice相等的定义我们选择最常见的需求，也就是当两个slice的类型和长度相同，且相等下标的值也是相等的，比如：上述代码中 ......

在golang中我们可以轻松地通过==来判断两个数组（array）是否相等，但遗憾的是slice并没有相关的运算符，当需要判断两个slice是否相等时我们只能另寻捷径了。

slice相等的定义

我们选择最常见的需求，也就是当两个slice的类型和长度相同，且相等下标的值也是相等的，比如：

a := []int{1, 2, 3}
b := []int{1, 2, 3}
c := []int{1, 2}
d := []int{1, 3, 2}

上述代码中a和b是相等的，c因为长度和a不同所以不相等，d因为元素的排列顺序和a不同所以也不相等。

判断两个[]byte是否相等

为什么要单独将[]byte列举出来呢？

因为标准库提供了优化的比较方案，不再需要我们造*了：

package main

import (
    "bytes"
    "fmt"
)

func main() {
    a := []byte{0, 1, 3, 2}
    b := []byte{0, 1, 3, 2}
    c := []byte{1, 1, 3, 2}

    fmt.println(bytes.equal(a, b))
    fmt.println(bytes.equal(a, c))
}

运行结果如下：

golang中判断两个slice是否相等

使用reflect判断slice是否相等

在判断类型不是[]byte的slice时，我们还可以借助reflect.deepequal，它用于深度比较两个对象包括它们内部包含的元素是否都相等：

func deepequal(x, y interface{}) bool

deepequal reports whether x and y are “deeply equal,” defined as follows. two values of identical type are deeply equal if one of the following cases applies. values of distinct types are never deeply equal.
...
slice values are deeply equal when all of the following are true: they are both nil or both non-nil, they have the same length, and either they point to the same initial entry of the same underlying array (that is, &x[0] == &y[0]) or their corresponding elements (up to length) are deeply equal. note that a non-nil empty slice and a nil slice (for example, []byte{} and []byte(nil)) are not deeply equal.

这段话的意思不难理解，和我们在本文最开始时讨论的如何确定slice相等的原则是一样的，只不过它借助了一点运行时的“黑魔法”。

看例子：

package main

import (
    "fmt"
    "reflect"
)

func main() {
    a := []int{1, 2, 3, 4}
    b := []int{1, 3, 2, 4}
    c := []int{1, 2, 3, 4}
    fmt.println(reflect.deepequal(a, b))
    fmt.println(reflect.deepequal(a, c))
}

golang中判断两个slice是否相等

手写判断

在golang中使用reflect通常需要付出性能代价，如果我们确定了slice的类型，那么自己实现slice的相等判断相对来说也不是那么麻烦：

func testeq(a, b []int) bool {
    // if one is nil, the other must also be nil.
    if (a == nil) != (b == nil) {
        return false;
    }

    if len(a) != len(b) {
        return false
    }

    for i := range a {
        if a[i] != b[i] {
            return false
        }
    }

    return true
}

测试代码：

package main

import "fmt"

func main() {
    a := []int{1, 2, 3, 4}
    b := []int{1, 3, 2, 4}
    c := []int{1, 2, 3, 4}
    fmt.println(testeq(a, b))
    fmt.println(testeq(a, c))
}

运行结果：

golang中判断两个slice是否相等

下面我们对后两种方案做个简单的性能测试，我们测试两个不相等但很相似的拥有20个元素的slice，这是在日常开发中较常见的情景：

func benchmarktesteq(b *testing.b) {
    a := []uint32{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20}
    c := []uint32{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 21}
    b.resettimer()
    for i := 0; i < b.n; i++ {
        _ = testeq(a, c)
    }
}

func benchmarkdeepequal(b *testing.b) {
    a := []uint32{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20}
    c := []uint32{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 21}
    b.resettimer()
    for i := 0; i < b.n; i++ {
        _ = reflect.deepequal(a, c)
    }
}

当然这个测试只能反应出有限的信息，正常情况下应该给出更全面的测试用例。不过在我们的演示中反射仍然付出了惊人的性能代价：

golang中判断两个slice是否相等