[Swift] 프로그래머스 행렬의 곱셈(lv. 2)

[Swift] 프로그래머스 행렬의 곱셈(lv. 2)

 

1. 문제

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

프로그래머스

 

2. 접근

행렬을 곱하라. 고등학교 이후로 오랜만에 보는 요구 조건이다. (요즘은 안배운다고...)

아무튼 곱셈을 할 수 있는 것만 주어지기 때문에, arr1의 원소들과 arr2의 각 원소의 n번째 원소들을 각각 곱하고 더해서... 새로운 행렬의 값들을 만들어내는 것이다.

 

k번째의 j번째 인덱스들을 서로 곱해서 그 합을 새로운 원소로 만들어야 하기 때문에 삼중 반복문을 써야 했다.

array를 초기화하는 등의 작업도 부가적으로 필요했다.

 

반복문 내부가 어떻게 돌아가지는지를 생각해보면서 하나 씩 만들어가면 그렇게 어렵지 않은 문제였다.

 

 

3. 코드

import Foundation

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