聲明：此博客是本人在泊學(xué)網(wǎng)學(xué)習(xí)Swift過(guò)程的筆記與心得

IV. 造輪子吧！

前面學(xué)習(xí)了Array的基本使用，不過(guò)，不了解實(shí)現(xiàn)機(jī)理還是不能更好的使用它，下面，就利用Swift的Extension功能，給Array添加自己的輪子吧！

??

下面的方法都是通過(guò)Extension進(jìn)行擴(kuò)展的，格式如下：

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

使用結(jié)果：

let testArr = [1, 2, 3, 4, 5] // 用于實(shí)現(xiàn)功能的測(cè)試數(shù)組

1.forEach -> myForEach

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

使用結(jié)果：

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

2.Map -> myMap

    // T 為泛型參數(shù)
    func myMap1<T>(_ method: (Element) -> T) -> [T] {
        var tmp: [T] = []
        for item in self {
            tmp.append(method(item))
        }
        
        return tmp
    }
    
    // 基于reduce實(shí)現(xiàn)，也可以替換成myReduce
    func myMap2<T>(_ method: (Element) -> T) -> [T] {
        return reduce([], { $0 + [method($1)] })
    }

使用結(jié)果：

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實(shí)現(xiàn)，也可以替換成myReduce
    func myFilter2(_ method: (Element) -> Bool) -> [Element] {
        return reduce([], { method($1) ? $0 + [$1]: $0 })
    }
    
    // 基于filter實(shí)現(xiàn)，也可以替換成myFilter
    func myReject(_ method: (Element) -> Bool) -> [Element] {
        return filter { !method($0) }
    }

使用結(jié)果：

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
    }

使用結(jié)果：

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
    }

使用結(jié)果：

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

6.allMatch -> myAllMatch

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

使用結(jié)果：

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
    }

使用結(jié)果：

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, "??")]

以上就是自己模擬系統(tǒng)實(shí)現(xiàn)的Array拓展，有時(shí)候，造一下輪子，可能有助于更好的理解系統(tǒng)的實(shí)現(xiàn)機(jī)制，更好的學(xué)習(xí)！