[Swift] 2차원 배열 90도 회전 코드

안녕하세요 ◠‿◠ 고고입니다.

// 2차원 배열 90도 회전하기
func rotate(a: [[Int]]) -> [[Int]] {
    let n = a.count // 행 길이 계산
    let m = a[0].count // 열 길이 계산
    var result = Array(repeating: Array(repeating: 0, count: m), count: n)
    for i in 0...n - 1 {
        for j in 0...m - 1 {
            result[j][n - i - 1] = a[i][j]
        }
    }
    return result
}