在 Go 中安全地对切片进行元素添加或删除操作

在 Go 中高效管理切片元素对于性能至关重要。本文探讨了查找和操作切片中特定元素的最佳方法，以避免逐一检查元素的低效做法。文章介绍了按字段排序和二分搜索、使用映射进行快速查找以及在切片元素添加或删除时同步映射和切片内容等技术。通过采用这些策略，开发者可以优化切片操作，提高代码效率，尤其是对于需要频繁查找和修改切片元素的场景。

假设我有 []struct{} 并且我需要知道切片中是否存在 id = A 的元素。如果存在，则根据用户输入的请求，该元素将被删除或移动到索引 0。那么，如何在不检查每个元素的情况下以最佳方式找到 golang 切片中的元素呢？或者，使用 slice.contains(obj) 就足够了吗？然后，如果该元素存在，我将根据用户输入中的请求执行操作。如果请求是 remove，我将在不破坏元素顺序的情况下删除它。但如果请求是add，我会将元素移动到索引0。

注意：该函数会经常被调用。 谢谢。

解决方案

编写通过迭代切片来查找元素的函数并不难：

func contains(s []your_struct, e int) (bool, int) {
    for idx, a := range s {
        if a.id == e {
            return true, idx
        }
    }
    return false, -1
}

如果您要经常调用该函数，则按 id 字段对切片进行排序并对 slice 的 your_struct 实现二分搜索可能会很有用。

如果切片不是很大，您可以创建额外的数据结构 - map[int]int 并在此映射中保留切片元素的索引。但在这种情况下，当您修改其中之一时，您需要同步切片和地图的内容：

your_map := make(map[int]int)
if idx, ok := your_map[id]; ok {
    // ...
}

如果需要多次检查的话 最好一次性创建 id 字段的 map[string]int 。

每次只需检查地图是否包含 id

这里，id作为键，切片index作为值

mp := make(map[string]int)
for idx, a := range yourstuctslice {
    mp[a.id] = idx
}
if idx, ok := mp[id]; ok {
    // remove the element using idx
}

如果切片中添加了新元素，则也会更新地图

mp[newelement.id] = true

如果要删除搜索到的元素，可以通过切片索引删除

func removeindex(s []yourstuct, index int) []int {
    return append(s[:index], s[index+1:]...)
}

if idx, ok := mp[id]; ok {
    yourStuctSlice = RemoveIndex(yourStuctSlice , idx)
     delete(mp , id); // Remove from map also for next search 
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《在 Go 中安全地对切片进行元素添加或删除操作》文章吧，也可关注golang学习网公众号了解相关技术文章。