Go语言桶排序索引越界问题：浮点数转整数异常求助

问题分析 #

你的代码触发索引越界panic，核心原因是浮点数精度误差：当元素等于maxValue时，理论上(v - minValue) / bucketSize的结果应为10.0，但浮点数计算的精度问题可能导致结果刚好等于10.0，直接转为int后得到10，超出了buckets的索引范围（0-9）。

解决方案 #

这里提供两种高效的修复方式：

方案1：直接处理最大值边界

在计算桶索引时，显式判断当前元素是否为最大值，若是则直接放入最后一个桶：

for _, v := range array {
    var bucketIdx int
    if v == maxValue {
        bucketIdx = bucketCount - 1 // 强制放入最后一个桶
    } else {
        bucketIdx = int((v - minValue) / bucketSize)
    }
    buckets[bucketIdx] = append(buckets[bucketIdx], v)
}

方案2：用math.Floor()取整并兜底

导入math包后，用向下取整确保索引不会超过预期范围，同时增加兜底判断避免极端精度问题：

import "math"

// ...

for _, v := range array {
    ratio := (v - minValue) / bucketSize
    bucketIdx := int(math.Floor(ratio))
    // 兜底处理：防止精度误差导致索引越界
    if bucketIdx >= bucketCount {
        bucketIdx = bucketCount - 1
    }
    buckets[bucketIdx] = append(buckets[bucketIdx], v)
}

修复后的完整代码 #

package main

import (
    "fmt"
    "math"
)

func bucketSort(array []float64) []float64 {
    // 获取数组最值
    minValue, maxValue := array[0], array[0]
    for _, v := range array {
        if v < minValue {
            minValue = v
        } else if v > maxValue {
            maxValue = v
        }
    }

    // 初始化桶
    bucketSize := (maxValue - minValue) / 10
    bucketCount := 10
    buckets := make([][]float64, bucketCount)

    // 分配元素到对应桶
    for _, v := range array {
        ratio := (v - minValue) / bucketSize
        bucketIdx := int(math.Floor(ratio))
        if bucketIdx >= bucketCount {
            bucketIdx = bucketCount - 1
        }
        buckets[bucketIdx] = append(buckets[bucketIdx], v)
    }

    // 排序桶内元素并合并
    i := 0
    for _, bucket := range buckets {
        insertionSort(bucket)
        for _, v := range bucket {
            array[i] = v
            i++
        }
    }

    return array
}

func insertionSort(array []float64) {
    n := len(array)
    for i := 1; i < n; i++ {
        value := array[i]
        j := i - 1
        for j >= 0 && array[j] > value {
            array[j+1] = array[j]
            j--
        }
        array[j+1] = value
    }
}

func main() {
    array := []float64{0.897, 0.565, 0.656, 0.1234, 0.665, 0.3434}
    fmt.Println("Sorted Array: ", bucketSort(array))
}

关键提示 #

• 浮点数运算的精度误差是数值排序场景的常见坑，必须显式处理边界值
• 两种方案各有优势：方案1代码更简洁，适合明确包含最大值的场景；方案2兼容性更强，能覆盖所有浮点数精度异常情况

内容的提问来源于stack exchange，提问作者Luke