登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  Golang >  Go教程

Go 泛型函数怎么给不同类型的切片去重

来源:17golang原创

时间:2026-09-06 06:23:10 473浏览 收藏

如果项目里同时有 []int[]string 和业务结构体切片,分别写去重函数很快就会变成重复劳动。这个问题适合交给 Go 泛型:让元素类型成为类型参数,用 map[T]struct{} 记录已经出现的值,再把第一次遇到的值按原顺序放入结果。

要点速览
  • T comparable 是用元素作为 map 键的必要约束。
  • 下面的实现平均时间复杂度为 O(n),并保持首次出现顺序。
  • 切片、map、func 不可比较,遇到这类元素应改用可比较的 key。
我们可以借助 Go 1.18 及之后版本支持的泛型特性,写出一套能兼容所有可比较类型的切片去重通用函数,不用再针对 int、string、自定义可比较结构体这些不同类型单独写重复的去重逻辑。
通过泛型限定元素为 comparable 可比较约束,用 map 存已经出现过的元素来去重,最终直接返回新的无重复切片,就可以一次性适配绝大多数常见类型切片的去重需求。

先把去重规则写成一个泛型函数

去重通常要同时满足两个条件:相同值只保留一次,而且结果顺序与输入中第一次出现的顺序一致。map[T]struct{} 适合做存在性集合,空结构体不携带额外数据;comparable 则告诉编译器,T 的值可以用于 ==!= 和 map 键。

package main

// Unique 保留每个元素第一次出现的位置,并删除后续重复值。
func Unique[T comparable](items []T) []T {
    // 用空结构体表示“已经见过”,避免为集合值分配无用数据。
    seen := make(map[T]struct{}, len(items))
    result := make([]T, 0, len(items))

    for _, item := range items {
        // map 查询只负责判断存在性,不需要取出一个业务值。
        if _, exists := seen[item]; exists {
            continue
        }
        seen[item] = struct{}{}
        result = append(result, item)
    }
    return result
}

容量按输入长度预留,避免结果不断扩容;它不会改变最终长度。函数没有修改原切片,所以调用方可以继续使用输入数据。

Go 泛型切片去重中的 comparable 元素进入 map[T]struct{} 集合并保留首次出现顺序
图1:T comparable 让元素进入存在性集合,结果切片只接收尚未出现的值。

用不同类型调用,类型参数通常可以自动推断

调用时不必把类型参数写在方括号里,编译器能从实参切片推断 T:

package main

import "fmt"

func main() {
    // 两种元素类型共用同一个去重实现。
    ids := Unique([]int{3, 7, 3, 9, 7})
    names := Unique([]string{"go", "rust", "go", "java"})

    fmt.Println(ids)   // [3 7 9]
    fmt.Println(names) // [go rust java]
}

如果上下文不足以推断类型,也可以显式写成 Unique[int](values)。对“可比较”的自定义结构体同样适用,例如只含整数、字符串、指针等可比较字段的 struct{ ID int; Name string }

comparable 不是“所有类型都能比较”

下面这张边界表很重要。数组和字段全可比较的结构体可以比较;切片、map、func 不能比较,因此不能直接作为 T

元素类型能否直接使用原因或替代方案
int、string、bool可以可比较,也能作为 map 键
数组、可比较结构体可以所有组成部分都必须可比较
[]byte、map、func不可以先把内容转成 string,或传入 key 函数
float64语法上可以NaN 不等于自身,需先定义业务规则

不要为了兼容不可比较元素而把约束改成 any:那样 map 键方案无法编译。更通用的设计是让调用方提供可比较键:

// UniqueBy 按调用方提供的键去重,适合结构复杂或不可比较的元素。
func UniqueBy[T any, K comparable](items []T, key func(T) K) []T {
    seen := make(map[K]struct{}, len(items))
    result := make([]T, 0, len(items))
    for _, item := range items {
        // 只把 K 放入集合,T 本身不需要满足 comparable。
        k := key(item)
        if _, exists := seen[k]; exists {
            continue
        }
        seen[k] = struct{}{}
        result = append(result, item)
    }
    return result
}
Go UniqueBy 用任意元素 T 映射到可比较键 K 后完成复杂结构切片去重
图2:当元素 T 不可比较时,只需把它映射到可比较键 K,去重规则仍保持清晰。

常见问题

泛型去重会不会自动排序?

不会。实现按输入遍历顺序追加结果,既不排序,也不依赖 map 的遍历顺序。

返回结果是 nil 还是空切片?

上面的实现对 nil 输入会创建长度为 0 的非 nil 结果。若接口需要保留 nil 语义,可在函数开头单独判断 items == nil

什么时候不该用 map 去重?

如果数据量很小、元素没有可比较键,或业务判断是“忽略大小写后相同”,应使用 UniqueBy 或先规范化键,再决定是否去重。

核心选择可以归纳为:元素本身可比较就用 Unique[T comparable];元素复杂但能提取稳定标识,就用 UniqueBy[T any, K comparable]。这样既复用实现,也把比较规则明确留在调用方。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>