为什么我的排序函数返回的值比输入的多答案

【问题标题】：Why is my sorting function returning more values than input为什么我的排序函数返回的值比输入的多
【发布时间】：2022-11-17 00:15:54
【问题描述】：

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

【问题讨论】：

你能澄清一下问题是什么吗？ sort_chunk_no_p() 用一个包含 100 个元素的切片调用，并返回另一个也包含 100 个元素的切片。
它被赋予 100 个元素，并返回 slice 的 slice 中的 100 多个元素。它正在创建 100 个切片，但我试图跳过之前已经分组的元素，这就是我试图使用布尔值的目的。 if !message1.Grouped 行似乎无关紧要。我会尝试在主帖正文中更好地澄清

标签： go

【解决方案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)

在Go Playground 上试试。

见相关：

Using Pointers in a for loop

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

【讨论】：