为什么这种 for 循环方法与 map 方法相比如此缓慢？

Question

我在 Playground 中测试了我的代码，但正如讨论所指出的那样，Playground 是调试配置，一旦我将所有这些代码放入真正的应用程序运行中，它们并没有太大的区别。以前不知道这个调试/发布的事情。

Swift 性能相关问题，我需要遍历图片的像素偏移量，首先我是这样尝试的。

func p1() -> [[Int]]{
    var offsets = [[Int]]()
    for row in 0..<height {
        var rowOffset = [Int]()
        for col in 0..<width {
            let offset = width * row + col
            rowOffset.append(offset)
        }
        offsets.append(rowOffset)
    }
    return offsets
}

但是很慢，我搜了一下发现有一段代码sn-p loop through offset这种方式：

func p2() -> [[Int]]{
    return (0..<height).map{ row in
        (0..<width).map { col in
            let offset = width * row + col
            return offset
        }
    }
}

所以我测试了如果我使用函数 p1 和 p2 循环通过 height = 128 和 width = 128 image ， p1 比 p2 慢 18 倍，为什么 p1 与 p2 相比如此慢？我也想知道这个任务还有其他更快的方法吗？

Answer 1

map 方法更快的最明显原因是因为map 预先分配了数组容量（因为它知道结果数组中有多少元素）。您也可以通过在您的数组上调用 ary.reserveCapacity(n) 在您的代码中执行此操作，例如

func p1() -> [[Int]]{
    var offsets = [[Int]]()
    offsets.reserveCapacity(height) // NEW LINE
    for row in 0..<height {
        var rowOffset = [Int]()
        rowOffset.reserveCapacity(width) // NEW LINE
        for col in 0..<width {
            let offset = width * row + col
            rowOffset.append(offset)
        }
        offsets.append(rowOffset)
    }
    return offsets
}