小编典典

如何优雅地检查三个值的相等性?

go

说我有价值abc。我想找出它们是否相等。如果我做

if a == b == c{...}

然后我得到一个编译错误

invalid operation: a == b == c (mismatched types bool and TypeOfABandC)

这很明显,因为它解析为:

(a == b) == c

并且(a == b)是布尔。

我当然可以做:

if a == b && a == c {...}

但是,这看起来不太好,让人感到困惑。还有另一种方法吗?


阅读 484

收藏
2020-07-02

共1个答案

小编典典

事先注意:

如果三个值相等, 最后提出的解决方案 最短,最清晰和最有效的比较方法:

if a == b && a == c {
    fmt.Println("Clearest: all 3 are equal")
}

或者(根据您的喜好):

if a == b && b == c {
    fmt.Println("Clearest: all 3 are equal")
}

这个答案的其余部分(以下内容)只是与语言规范和语言功能有关,展现了我发现的有趣和创造性。他们没有尝试提供出色的解决方案。


GoPlayground上尝试以下所有示例。这些示例基于Spec:比较运算符中定义的条件和比较结果。

一般说明:在下面的示例中,我使用了可以在interface{}您的值具有(abc)的任何类型下工作的类型,但是int,例如,如果您知道它们的类型,则也可以使用该特定类型(这将提高效率并缩短示例的长度)。

map一套

if len(map[interface{}]int{a: 0, b: 0, c: 0}) == 1 {
    fmt.Println("Map set: all 3 are equal")
}

实际上,我们将所有可比较的值都放在a中map作为键,如果所有值相等,则映射中将只有1对,因此映射的“长度”将为1。映射的值类型不会发挥任何作用在这里扮演角色,可能是任何事情。我int之所以使用,是因为这会导致最短的复合文字(定义map值)。

该解决方案非常灵活,因为您可以使用任意数量的值来测试所有值是否相等,而不仅仅是3。

带阵列

数组是可比较的(与切片不同):

if [2]interface{}{a, b} == [2]interface{}{b, c} {
    fmt.Println("Arrays: all 3 are equal")
}

数组比较的结果将是trueif a == bb == c(如果相应的元素相等)。

请注意,您也可以将此方法应用于任意数量的值。对于5个值,它看起来像这样:

if [4]interface{}{a, b, c, d} == [4]interface{}{b, c, d, e} {
    fmt.Println("Arrays: all 5 are equal")
}

棘手的 map

if map[interface{}]bool{a: b == c}[b] {
    fmt.Println("Tricky map: all 3 are equal")
}

该复合文字将把比较结果分配给b == ca。然后我们询问与键关联的值b。如果a == b为,则索引表达式的结果将为的结果b == c,即所有3个值是否相等。如果为a != b,则将是值类型的零值(false在的情况下),这将bool正确地表明所有3个值都不相等。

与匿名structs

struct 值也具有可比性,因此:

if struct{ a, b interface{} }{a, b} == struct{ a, b interface{} }{b, c} {
    fmt.Println("Anon structs: all 3 are equal")
}

与(named)structs

如果我们struct事先定义一个简单的方法:

type P struct{ a, b interface{} }

比较将更加紧凑:

if (P{a, b} == P{b, c}) {
    fmt.Println("Structs: all 3 are equal")
}

请注意,该if语句的表达式必须放在括号中以避免[解析歧义 -否则,这是编译时错误!)

带片

切片无法比较,因此我们需要向借些帮助reflect.DeepEqual()

if reflect.DeepEqual([]interface{}{a, b}, []interface{}{b, c}) {
    fmt.Println("Slices: all 3 are equal")
}

具有辅助功能

func AllEquals(v ...interface{}) bool {
    if len(v) > 1 {
        a := v[0]
        for _, s := range v {
            if a != s {
                return false
            }
        }
    }
    return true
}

并使用它:

if AllEquals(a, b, c) {
    fmt.Println("Helper function: all 3 are equal")
}

该解决方案也很灵活,可以调用AllEquals()任意数量的值。

请注意,AllEquals()如果我们也愿意在reflect.DeepEqual()这里调用,可以使函数更紧凑:

func AllEquals2(v ...interface{}) bool {
    if len(v) < 2 {
        return true
    }
    return reflect.DeepEqual(v[:len(v)-1], v[1:])
}
2020-07-02