Get item with row and col from 1d array

1dArrayWithColRow.swift

extension Collection where Self.Index == Int {
    subscript(row: Int, col: Int, colCount: Int) -> Element? {
        let index = row * colCount + col
        guard index >= 0, index < count else { return nil }
        return self[index]
    }
}