巧用 Go Map 特性对数组或切片去重

简介: 本文介绍了如何利用 Go 的复合数据类型 Map 的特性对数组或切片进行去重。值得注意的一个地方是,在使用 Map 构建 Set 时,Value 的数据类型指定为 struct{},原因是后面在添加键值对的时候,指定的 Value 为空结构体 strcut{}{},空结构体不占用内存空间。

耐心和持久胜过激烈和狂热。

哈喽大家好,我是陈明勇,本文分享的内容是巧用 Go Map 特性对数组或切片进行去重。如果本文对你有帮助,不妨点个赞,如果你是 Go 语言初学者,不妨点个关注,一起成长一起进步,如果本文有错误的地方,欢迎指出!

前言

在其他语言里(如 Java )使用过 Set 集合的小伙伴都知道,它的一个特点就是集合里的元素是不重复的。在一些需要去重场景中,我们可以使用 Set 这种数据结构去存储数据,达到去重的目的。而在 Go 语言里面,是没有 Set 这种数据类型的。如果我们想利用无重复元素的数据结构对数组或切片去重,可以使用 Go 的复合数据类型 Map 去构建 Set 集合,对数组或切片进行去重。因为 Map 中的 key 是唯一的,如果不了解 Map 的小伙伴,可以看看我之前的文章。

使用 Map 构建 Set

type MySet map[any]struct{}
复制代码

定义一个自定义类型 MySet ,具有 map[any]strcut{} 的特性。

  • mapkeyany 类型,任何类型都可以当做 mapkey
  • mapvaluestruct{} 类型,为什么会用结构体类型?是因为后面添加 k-v 键值对时,指定的 value 为 空结构体 strcut{}{}fmt.Println(unsafe.Sizeof(struct{}{})) // 0 空结构体是不占用内存空间的。

对数组或切片去重

import "fmt"
type MySet map[any]struct{}
func main() {
  s1 := []int{1, 1, 2, 3}
  fmt.Println(duplicateRemoving(s1)) // [1 2 3]
  s2 := []string{"a", "b", "c", "a"}
  fmt.Println(duplicateRemoving(s2)) // [a b c]
}
func duplicateRemoving[T any](s []T) []T {
  res := make([]T, 0, len(s))
  mySet := make(MySet)
  for _, t := range s {
    if _, ok := mySet[t]; !ok {
      res = append(res, t)
      mySet[t] = struct{}{}
    }
  }
  return res
}
复制代码

核心思路

  • 定义一个新的切片 res,为去重后的切片;定义一个 MySet 类型的变量 mySet,用于判断元素是否重复;
  • 遍历原切片 s,首先判断 mySet 里是否存在以切片 s 的元素 tkey 的键值对,如果不存在,往新切片 res 添加元素 t,往 mySet 里添加键值对 t → struct{}{},否则说明元素重复,不做添加操作;
  • 返回去重后的新切片。

小结

本文介绍了如何利用 Go 的复合数据类型 Map 的特性对数组或切片进行去重。值得注意的一个地方是,在使用 Map 构建 Set 时,Value 的数据类型指定为 struct{},原因是后面在添加键值对的时候,指定的 Value 为空结构体 strcut{}{},空结构体不占用内存空间。

目录
相关文章
|
16天前
|
Go
Go 中使用切片来实现动态数组的功能
Go 中使用切片来实现动态数组的功能
|
20天前
|
存储 安全 测试技术
【Go语言精进之路】构建高效Go程序:了解map实现原理并高效使用
【Go语言精进之路】构建高效Go程序:了解map实现原理并高效使用
29 3
|
20天前
|
存储 监控 Go
【Go语言精进之路】构建高效Go程序:了解切片实现原理并高效使用
【Go语言精进之路】构建高效Go程序:了解切片实现原理并高效使用
26 3
|
12天前
|
Go
go语言map、实现set
go语言map、实现set
14 0
|
12天前
|
Go
go语言数组与切片
go语言数组与切片
17 0
|
1天前
|
JavaScript API
js【最佳实践】遍历数组的八种方法(含数组遍历 API 的对比)for,forEach,for of,map,filter,reduce,every,some
js【最佳实践】遍历数组的八种方法(含数组遍历 API 的对比)for,forEach,for of,map,filter,reduce,every,some
8 1
|
16天前
|
缓存 安全 算法
Go 中使用 map 实现高效的数据缓存
Go 中使用 map 实现高效的数据缓存
|
16天前
|
存储 缓存 安全
Go 中使用 map 实现高效的数据查找和更新
Go 中使用 map 实现高效的数据查找和更新
|
16天前
|
Go
Go 语言是如何实现切片扩容
Go 语言是如何实现切片扩容
|
12天前
|
Go
go切片和map比较
go切片和map比较
11 0