Натхненний https://www.swiftbysundell.com/posts/the-power-of-key-paths-in-swift , ми можемо оголосити більш потужний інструмент, який здатний фільтрувати за єдиність на будь-якому keyPath. Завдяки коментарям Олександра до різних відповідей щодо складності, наведені нижче рішення повинні бути майже оптимальними.
Немутуючий розчин
Ми розширюємо функцію, яка здатна фільтрувати єдиність на будь-якому keyPath:
extension RangeReplaceableCollection {
/// Returns a collection containing, in order, the first instances of
/// elements of the sequence that compare equally for the keyPath.
func unique<T: Hashable>(for keyPath: KeyPath<Element, T>) -> Self {
var unique = Set<T>()
return filter { unique.insert($0[keyPath: keyPath]).inserted }
}
}
Примітка: у випадку, коли ваш об'єкт не відповідає RangeReplaceableCollection, але відповідає послідовності, ви можете мати це додаткове розширення, але типом повернення завжди буде масив:
extension Sequence {
/// Returns an array containing, in order, the first instances of
/// elements of the sequence that compare equally for the keyPath.
func unique<T: Hashable>(for keyPath: KeyPath<Element, T>) -> [Element] {
var unique = Set<T>()
return filter { unique.insert($0[keyPath: keyPath]).inserted }
}
}
Використання
Якщо ми хочемо єдиності для самих елементів, як у питанні, ми використовуємо keyPath \.self
:
let a = [1, 4, 2, 2, 6, 24, 15, 2, 60, 15, 6]
let b = a.unique(for: \.self)
/* b is [1, 4, 2, 6, 24, 15, 60] */
Якщо ми хочемо єдиності для чогось іншого (наприклад, для id
колекції об’єктів), тоді ми використовуємо наш вибір KeyPath:
let a = [CGPoint(x: 1, y: 1), CGPoint(x: 2, y: 1), CGPoint(x: 1, y: 2)]
let b = a.unique(for: \.y)
/* b is [{x 1 y 1}, {x 1 y 2}] */
Мутуючий розчин
Ми розширюємо функцію, що мутує, яка здатна фільтрувати єдиність на будь-якому keyPath:
extension RangeReplaceableCollection {
/// Keeps only, in order, the first instances of
/// elements of the collection that compare equally for the keyPath.
mutating func uniqueInPlace<T: Hashable>(for keyPath: KeyPath<Element, T>) {
var unique = Set<T>()
removeAll { !unique.insert($0[keyPath: keyPath]).inserted }
}
}
Використання
Якщо ми хочемо єдиності для самих елементів, як у питанні, ми використовуємо keyPath \.self
:
var a = [1, 4, 2, 2, 6, 24, 15, 2, 60, 15, 6]
a.uniqueInPlace(for: \.self)
/* a is [1, 4, 2, 6, 24, 15, 60] */
Якщо ми хочемо єдиності для чогось іншого (наприклад, для id
колекції об’єктів), тоді ми використовуємо наш вибір KeyPath:
var a = [CGPoint(x: 1, y: 1), CGPoint(x: 2, y: 1), CGPoint(x: 1, y: 2)]
a.uniqueInPlace(for: \.y)
/* a is [{x 1 y 1}, {x 1 y 2}] */
NSSet
, NSSet - це не упорядкована колекція об'єктів, якщо потрібно підтримувати порядок NSOrderedSet.