【Swift】 Collections初识 - Array(三)

声明：此博客是本人在泊学网学习Swift过程的笔记与心得

IV. 造轮子吧！

前面学习了Array的基本使用，不过，不了解实现机理还是不能更好的使用它，下面，就利用Swift的Extension功能，给Array添加自己的轮子吧！

下面的方法都是通过Extension进行扩展的，格式如下：

extension Array {
    func function(_ xxx: X) -> xxx {}
}

使用结果：

let testArr = [1, 2, 3, 4, 5] // 用于实现功能的测试数组

1.forEach -> myForEach

    func myForEach(_ method: (Element) -> ()) -> () {
        for item in self {
            method(item)
        }
    }

使用结果：

testArr.myForEach { print($0) }
// 1
// 2
// 3
// 4
// 5

2.Map -> myMap

    // T 为泛型参数
    func myMap1<T>(_ method: (Element) -> T) -> [T] {
        var tmp: [T] = []
        for item in self {
            tmp.append(method(item))
        }

        return tmp
    }

    // 基于reduce实现，也可以替换成myReduce
    func myMap2<T>(_ method: (Element) -> T) -> [T] {
        return reduce([], { $0 + [method($1)] })
    }

使用结果：

let arrMap1 = testArr.myMap1 { "\($0 * $0)" }
let arrMap2 = testArr.myMap2 { "\($0 * $0)" }
arrMap1 // ["1", "4", "9", "16", "25"]
arrMap2 // ["1", "4", "9", "16", "25"]

3.filter -> myFilter 、myReject(反向筛选)

    func myFilter1(_ method: (Element) -> Bool) -> [Element] {
        var tmp: [Element] = []
        for item in self where method(item) {
            tmp.append(item)
        }

        return tmp
    }

    // 基于reduce实现，也可以替换成myReduce
    func myFilter2(_ method: (Element) -> Bool) -> [Element] {
        return reduce([], { method($1) ? $0 + [$1]: $0 })
    }

    // 基于filter实现，也可以替换成myFilter
    func myReject(_ method: (Element) -> Bool) -> [Element] {
        return filter { !method($0) }
    }

使用结果：

let arrFilter1 = testArr.myFilter1 { $0 % 2 == 0 }
let arrFilter2 = testArr.myFilter2 { $0 % 2 == 0 }
let arrReject = testArr.myReject { $0 % 2 == 0 }
arrFilter1 // [2, 4]
arrFilter2 // [2, 4]
arrReject // [1, 3, 5]

4.reduce -> myReduce

    func myReduce<T>(_ first: T, _ method: (T, Element) -> T) -> T {
        var tmp: T = first

        for item in self {
            tmp = method(tmp, item)
        }

        return tmp
    }

使用结果：

let arrReduce = testArr.myReduce(0, +) 
arrReduce // 15

5.contains -> myContains

    func myContains(_ method: (Element) -> Bool) -> Bool {
        for item in self where method(item) {
            return true
        }

        return false
    }

使用结果：

let arrContains = testArr.myContains { $0 / 2 == 2 }
arrContains // true

6.allMatch -> myAllMatch

    func myAllMatch(_ method: (Element) -> Bool) -> Bool {
        return !contains { !method($0) }
    }

使用结果：

let arrAllMatch = testArr.myAllMatch { $0 % 2 == 0 }
arrAllMatch // false

7.flatMap -> myFlatMap

    func myFlatMap<T>(_ method: (Element) -> [T]) -> [T] {
        var tmp: [T] = []

        for item in self {
            tmp.append(contentsOf: method(item))
        }

        return tmp
    }

使用结果：

let arrFlatMap = testArr.myFlatMap { id in
    return ["狗", "猫", "熊"].map { animal in
        return (id, animal)
    }
}
arrFlatMap // [(1, "狗"), (1, "猫"), (1, "熊"), (2, "狗"), (2, "猫"), (2, "熊"), (3, "狗"), (3, "猫"), (3, "熊"), (4, "狗"), (4, "猫"), (4, "熊"), (5, "狗"), (5, "猫"), (5, "熊")]

以上就是自己模拟系统实现的Array拓展，有时候，造一下轮子，可能有助于更好的理解系统的实现机制，更好的学习！