Swift:移除数组中的重复元素

引子

从数组中去除重复项用来获取唯一值,在日常开发中是一项常见的操作。像 Ruby 这样的语言有内置的 uniq 方法,但在 Swift 中,我们必须自己创建这样的方法。标准库并没有提供一个简单的实现此功能的方法。

有多种方法可以达到相同的结果,每种方法都有其自身的优点和缺点。让我们一起来探讨一下,看看哪种方法最适合你的应用场景。

通过 Set 来去除重复元素

在我们开始探讨如何对数组中的重复项进行删除的扩展操作之前,先了解一下 Swift 中的 Set 是个不错的选择。默认情况下,集合只会包含唯一的值,这样就可以满足我们需要的效果。

如果我们选择使用 Set 来实现的话,有两个注意点:

  • 我们不需要元素有序。
  • 每个元素都是唯一的。
swift 复制代码
let arrayOfnums: [Int] = [1, 1, 2, 2, 3, 3]
let setOfNums: Set<Int> = [1, 1, 1, 2, 2, 2, 3, 3, 3]

print(arrayOfnums) // [1, 1, 2, 2, 3, 3]
print(setOfNums)   // [2, 3, 1]

这更多地属于一种代码设计决策,即你需要选择使用集合(Set)还是数组(Array)。集合的一个优点是性能更优,这使得它们成为解决唯一性问题的绝佳选择。

无需过多详述其差异所在,但要知道集合并不会保持元素的顺序。如果在你的需求是必须保持元素顺序,那么你可能就不应选择使用集合。

你可以选择使用 NSOrderedSet,但这个类无法提供类型补全功能,你将不得不处理 Any 的实例。

通过给数组添加扩展函数来实现去重

当元素的顺序至关重要时,我们可以继续使用数组,并通过使用自定义扩展来获取唯一的值。

如果要使用数组的话:

  • 顺序很重要
  • 你不能轻易切换到一个集合

我们需要创建一个扩展来过滤掉重复的元素。在设计这个扩展时,务必考虑到性能问题,因为如果我们不注意这一点,很快就会出现 O(N²) 的二次时间复杂度。

简而言之,这意味着你拥有的元素越多,性能就会越差。以下这段代码依赖于 Hashable 协议来匹配元素,并且其时间复杂度为线性的 O(N)。这意味着 3 个元素需要 3 次迭代,10 个元素需要 10 次迭代,以此类推。示例代码如下:

dart 复制代码
extension Sequence where Iterator.Element: Hashable {
    func unique() -> [Iterator.Element] {
        var seen: Set<Iterator.Element> = []
        return filter { seen.insert($0).inserted }
    }
}

print(arrayOfnums.unique()) // [1, 2, 3]

让我们来剖析一下这个 unique() 方法:

  • 我们创建一个集合来跟踪已查看的对象
  • 使用一个过滤器来遍历所有对象
  • insert(_:) 方法会返回一个包含一个插入布尔值的元组,如果对象被插入则该布尔值设为 true,否则设为 false
  • 插入的布尔值值用于从我们的数组中过滤掉重复项

最终结果是一个顺序相同但没有重复元素的数组。唯一的缺点是数组的元素需要符合 Hashable 协议,但这应该不是什么大问题。标准库中的许多类型(如字符串、整数和布尔值)都已符合该协议。Hashable 用于确定一个元素是否与现有对象相等,因此是获取唯一值所必需的。

相关推荐
songgeb16 小时前
Concurrency in Swift学习笔记-初识
ios·swift
杂雾无尘1 天前
2025 年了,是否该全面拥抱 Swift 6?
ios·swift·客户端
杂雾无尘2 天前
用高斯公式优化 Swift 代码,让运行速度飞跃数十万倍!
ios·swift·apple
杂雾无尘3 天前
解决 Xcode 烦人错误:"Build input file cannot be found" 一招搞定!
ios·swift·客户端
代码小学僧3 天前
前端开发切图利器!Exportx,欢迎大家使用体验
前端·设计·掘金·金石计划
水木姚姚4 天前
图书管理软件iOS(iPhone)
macos·ios·iphone·xcode·swift
东坡肘子4 天前
F1:电影很好看,赛事很挣钱 | 肘子的 Swift 周报 #094
swiftui·swift·apple
YungFan5 天前
iOS26适配指南之动画
ios·swift
瓜子三百克6 天前
Swift6.1 - 基础知识1: 简单值、控制流、函数和闭包
开发语言·swift
songgeb6 天前
Currying and Partial application
swift·函数式编程