Go:附加,如果唯一

Question

有没有办法检查切片/贴图是否存在值？

我想仅在切片中不存在切片时才向切片添加值.

这有效,但似乎很冗长.有没有更好的方法来做到这一点？

orgSlice := []int{1, 2, 3}
newSlice := []int{}
newInt := 2

newSlice = append(newSlice, newInt)
for _, v := range orgSlice {
    if v != newInt {
        newSlice = append(newSlice, v)
    }
}

newSlice == [2 1 3]

Answer 1

每次插入时,您的方法都需要线性时间.更好的方法是使用a map[int]struct{}.或者,您也可以使用map[int]bool或类似的东西,但空struct{}的优点是它不占用任何额外的空间.因此map[int]struct{}是一组整数的流行选择.

例:

set := make(map[int]struct{})
set[1] = struct{}{}
set[2] = struct{}{}
set[1] = struct{}{}
// ...

for key := range(set) {
  fmt.Println(key)
}
// each value will be printed only once, in no particular order


// you can use the ,ok idiom to check for existing keys
if _, ok := set[1]; ok {
  fmt.Println("element found")
} else {
  fmt.Println("element not found")
}

Answer 2

如果你没有找到它,效率最高的可能是在切片上迭代并附加.

func AppendIfMissing(slice []int, i int) []int {
    for _, ele := range slice {
        if ele == i {
            return slice
        }
    }
    return append(slice, i)
}

它简单明了,对于小型列表来说速度很快.

此外,它总是比您当前基于地图的解决方案更快.无论如何,基于地图的解决方案遍历整个切片; 当此解决方案发现新值已存在时立即返回.两种解决方案都会在迭代时比较元素.(每个地图赋值语句当然在内部至少进行一次地图键比较.)只有在可以跨多次插入维护地图时,地图才有用.如果您在每次插入时重建它,那么所有优势都将丢失.

如果您确实需要有效处理大型列表,请考虑按排序顺序维护列表.(我怀疑顺序对你没关系,因为你的第一个解决方案附加在列表的开头,最后的解决方案附加在最后.)如果你始终保持列表排序,那么你可以使用sort.Search函数来做有效的二进制插入.