数据结构和算法分析LeetCode

867. Transpose Matrix

2018-09-16  本文已影响1人  楷书

矩阵转换,注意下标索引的正确映射。

class Solution {
    func transpose(_ A: [[Int]]) -> [[Int]] {
        guard A.count >= 1 && A[0].count >= 1 else {
            return [[Int]]()
        }
        
        var a: [Int] = Array(repeating: 0, count: A.count)
        var ret: [[Int]] = Array(repeating: a, count: A[0].count)
        
        for j in 0..<A[0].count {
            for i in 0..<A.count {
                ret[j][i] = A[i][j]
            }
        }
        
        return ret
    }
}
上一篇 下一篇

猜你喜欢

热点阅读