集合类型扩展

集合类型扩展是一种在编程中使用集合类型的方法，它可以扩展集合类型的功能，使其更加适用于特定的应用场景。现在让我们来看看一些常见的集合类型扩展。

数组扩展

在 Swift 中，数组是一种基本的集合类型。可以通过以下方式扩展数组的功能。

扩展添加元素的方法

我们可以通过自定义 append、insert 和 remove 方法来扩展数组添加和删除元素的功能。这样我们就可以通过自己的扩展方法来实现数组的特殊功能。

extension Array {
    mutating func appendUnique(_ element: Element) {
        if !contains(element) {
            append(element)
        }
    }
}

扩展快捷访问方法

我们可以通过自定义下标来快速访问数组中的元素。我们也可以通过下标来设置数组中的元素。例如，你可以通过下标的方式访问数组最后一个元素。

extension Array {
    var last: Element {
        return self[count - 1]
    }
    
    mutating func replaceLast(_ element: Element) {
        self[count - 1] = element
    }
}

字典扩展

字典是一种键值对的集合类型。需要注意的是，Swift 中的字典是无序的。下面是用于扩展字典的方法。

扩展添加元素的方法

我们可以通过自定义 updateValue 方法向字典中添加新元素。这个方法将值添加到字典中，如果不存在对应键值对，则新增一个键值对。如果已经存在对应的键值对，则更新对应的值。

extension Dictionary {
    mutating func addOrUpdateValue(_ value: Value, forKey key: Key) {
        updateValue(value, forKey: key)
    }
}

扩展对字典中元素的操作方法

我们可以通过自定义 findIndex 方法来查找某个值在字典中的索引位置，通过 removeAtIndex 方法来删除某个索引位置上的键值对。

extension Dictionary {
    func findIndex(key: Key) -> Int? {
        return Array(keys).firstIndex(of: key)
    }
    
    mutating func removeAtIndex(index: Int) {
        let key = Array(keys)[index]
        removeValue(forKey: key)
    }
}

集合类型的扩展是非常有用的，因为它可以使我们更加高效和便捷地处理各种数据类型。一个好的扩展可以简化代码的逻辑，提高代码的可读性和可维护性。