프로그래머스 - 행렬의 곱셈(Swift)

프로그래머스 - 행렬의 곱셈(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

 

 나의 풀이

 

행렬의 곱셈은 a행렬과 b행렬이 있을때 a행렬의 행과 b행렬의 열의 개수가 같아야 곱해질 수 있다.

그리고 결과 행렬은 a의 행의 개수, b의 열의 개수를 따르게 된다.

 

import Foundation

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    
    var arr3: [[Int]] = [[Int]](repeating: [Int](repeating: 0, count: arr2[0].count), count: arr1.count)
    
    for i in 0..<arr1.count {
        for j in 0..<arr2[0].count {
            var sum = 0
            for k in 0..<arr1[i].count {
                sum += arr1[i][k] * arr2[k][j]
            }
            arr3[i][j] = sum    
        }
    }
    return arr3
}

 

 

 

 다른 사람의 풀이

 

Array타입 안에 [Int]()를 통해서 2차원 배열을 초기화 할 수 있었다.

 

import Foundation

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    
    var arr3 = Array(repeating: [Int](), count: arr1.count)
    
    for i in 0..<arr1.count {
        for j in 0..<arr2[0].count {
            var sum = 0
            for k in 0..<arr1[0].count {
                sum += arr1[i][k] * arr2[k][j]
            }
            arr3[i].append(sum)
        }
    }
    
    return arr3
}

 

 피드백

 

행렬의 곱셈의 결과는 a, b일때 a의 행의 개수, b의 열의 개수로 구성된다.

arr1이라는 2차원배열의 열의 개수는 arr1[0].count이다.