为什么我的排序函数返回的值多于输入值

Question

我在下面发布的代码是最低可重现版本，因为我一直在尝试找出问题所在。我来自 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)

}

Answer 1

当您迭代一个切片时，所有元素都被复制到一个单独的、可重复使用的循环变量中。这意味着如果您修改此副本的字段，则切片中的元素不会受到影响。要么将指针存储在切片中（元素仍将被复制，但元素现在将是指向相同结构值的指针），要么通过索引表达式修改元素，例如chunk[i].Grouped = true.使用指针这就是它的样子：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)