排序函数输出结果数量超出了输入数量的原因

哈喽！今天心血来潮给大家带来了《排序函数输出结果数量超出了输入数量的原因》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

我在下面发布的代码是最小的可重现版本，因为我一直在尝试隔离问题。我来自 python，出于性能原因需要用 go 重写此脚本，特别是使用我从示例中删除的并行化。

问题是我将 n 个值传递给排序函数并获得 >n 返回值。它为第一个外循环中的每次迭代创建一个新切片，并且似乎忽略 if !message1.grouped 条件。我对 go 没有太多经验，但我用 python 来完成这个工作。我假设它与设置 message2.grouped = true 出于某种原因未被外循环看到有关。 最终我试图忽略已经在循环中先前分组的“消息”。

旁注：我知道这个脚本中的随机不起作用，因为我没有设置新的种子，但这不是重点，也不是我实际脚本的一部分

package main
import (
    "fmt"
    "math/rand"
)

type (
    BoolInt struct {
        Val int
        Grouped bool
    }
)


func sort_chunk_no_p(chunk []BoolInt) [][]BoolInt {
    COSINE_THRESHOLD := 0.90
    allGroups := [][]BoolInt{}
    for i, message1 := range chunk {
        if !message1.Grouped {
            message1.Grouped = true
            tempGroup := []BoolInt{message1}
            for _, message2 := range chunk[i+1:] {
                if !message2.Grouped {
                    if rand.Float64() >= COSINE_THRESHOLD {
                        message2.Grouped = true
                        tempGroup = append(tempGroup, message2)
                    }   
                }

            }
            allGroups = append(allGroups, tempGroup)
        }
    }
    return allGroups
}

func main() {
    lo, hi := 1, 100
    allMessages := make([]BoolInt, hi-lo+1)
    for i := range allMessages {
        allMessages[i].Val = i + lo
        allMessages[i].Grouped = false
    }

    sorted_chunk := sort_chunk_no_p(allMessages)


    fmt.Println(sorted_chunk)
    sum := 0
    for _, res := range sorted_chunk {
        sum += len(res)
    }
    fmt.Println(sum)
}

正确答案

当您迭代切片时，所有元素都将复制到单个重复使用的循环变量中。这意味着如果您修改此副本的字段，切片中的元素不会受到影响。

要么在切片中存储指针（元素仍将被复制，但元素现在将是指向相同结构值的指针），要么通过索引表达式（例如 chunk[i].grouped = true）修改元素。 p>

使用指针，这就是它的样子：

func sort_chunk_no_p(chunk []*boolint) [][]*boolint {
    cosine_threshold := 0.90
    allgroups := [][]*boolint{}
    for i, message1 := range chunk {
        if !message1.grouped {
            message1.grouped = true
            tempgroup := []*boolint{message1}
            for _, message2 := range chunk[i+1:] {
                if !message2.grouped {
                    if rand.float64() >= cosine_threshold {
                        message2.grouped = true
                        tempgroup = append(tempgroup, message2)
                    }
                }

            }
            allgroups = append(allgroups, tempgroup)
        }
    }
    return allgroups
}

并调用它：

allMessages := make([]*BoolInt, hi-lo+1)
for i := range allMessages {
    allMessages[i] = &BoolInt{Val: i + lo}
}

sorted_chunk := sort_chunk_no_p(allMessages)

在 Go Playground 上尝试一下。

查看相关内容：

Using Pointers in a for loop

Register multiple routes using range for loop slices/map

Why do these two for loop variations give me different behavior?

理论要掌握，实操不能落！以上关于《排序函数输出结果数量超出了输入数量的原因》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！