在Go语言的实际开发场景中,经常需要判断一个整数切片是否是另一个整数切片的子集,很多基础实现只考虑了元素是否存在,没有处理重复元素的情况,比如目标切片有两个元素5,待判断切片只有一个元素5时,基础方案会错误判定为子集。我们需要一种能兼顾重复元素的通用判断方案。

传统无重复判断方案的局限性
最常见的子集判断思路是将目标切片的元素存入哈希集合,再遍历待判断切片检查每个元素是否在集合中。这种方案只能判断元素是否存在,无法处理重复元素的频次问题。比如目标切片是[1,2,2,3],待判断切片是[1,2,3],按照该方案会判定为子集,但实际上待判断切片缺少一个2,不符合子集要求。
基于频次统计的实现思路
要解决重复元素的问题,核心是对两个切片的元素频次分别统计,再对比每个元素的频次:如果待判断切片中每个元素的出现次数都小于等于目标切片中对应元素的出现次数,那么待判断切片就是目标切片的子集。
具体实现步骤可以分为三步:
- 遍历目标切片,统计每个元素的出现频次,存入map[int]int结构
- 遍历待判断切片,统计每个元素的出现频次,同样存入map[int]int结构
- 遍历待判断切片的频次map,对比每个元素的频次是否小于等于目标切片的对应频次,只要有一个不满足就返回false,全部满足则返回true
完整代码实现
下面是兼顾重复元素的整数切片子集判断的完整Go语言实现:
package main
import "fmt"
// IsSubset 判断sub切片是否是mainSlice的子集,兼顾重复元素
func IsSubset(mainSlice []int, sub []int) bool {
// 统计主切片的元素频次
mainCount := make(map[int]int)
for _, num := range mainSlice {
mainCount[num]++
}
// 统计待判断切片的元素频次
subCount := make(map[int]int)
for _, num := range sub {
subCount[num]++
}
// 对比频次
for num, cnt := range subCount {
if mainCount[num] < cnt {
return false
}
}
return true
}
func main() {
// 测试用例1:包含重复元素,sub缺少重复元素
main1 := []int{1, 2, 2, 3}
sub1 := []int{1, 2, 3}
fmt.Println(IsSubset(main1, sub1)) // 输出 false
// 测试用例2:sub重复元素数量等于主切片
main2 := []int{1, 2, 2, 3}
sub2 := []int{1, 2, 2, 3}
fmt.Println(IsSubset(main2, sub2)) // 输出 true
// 测试用例3:sub包含主切片没有的元素
main3 := []int{1, 2, 3}
sub3 := []int{1, 2, 4}
fmt.Println(IsSubset(main3, sub3)) // 输出 false
// 测试用例4:空切片场景
main4 := []int{1, 2, 3}
sub4 := []int{}
fmt.Println(IsSubset(main4, sub4)) // 输出 true
}
性能优化建议
如果待判断切片的长度远大于目标切片,可以先做长度校验:如果待判断切片的长度大于目标切片的长度,直接返回false,避免不必要的频次统计操作,提升判断效率。
另外,如果两个切片都是有序的,也可以先排序再双指针遍历对比,不需要额外的map空间,不过排序的时间复杂度是O(n log n),而map方案的时间复杂度是O(n),实际使用时可以根据数据特点选择。
总结
通过元素频次统计的方式,可以完美解决Go语言中整数切片子集判断的重复元素问题,该方案逻辑清晰,时间复杂度为O(n),空间复杂度为O(n),适合大部分业务场景使用。如果场景中对空间要求更高且切片有序,也可以选择排序加双指针的优化方案。