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

코딩테스트 연습 - 행렬의 곱셈

Foma's 풀이

 

arr1은 행으로 각 배열의 숫자를 가로방향(→)으로 이동하고 다음 배열로 이동합니다.

[2,3,2]

[4,2,4]

[3,1,4]

arr2 열으로 각 배열을 세로방향(↓)으로 이동하고 배열안의 다음 숫자로 이동합니다.

[5,4,3]

[2,4,1]

[3,1,1]

arr1의 행과 arr2의 열을 각각 곱해준 값을 더한값을 answer에 차례대로 담아줍니다.

ex)

2 * 5 + 3 * 2 + 2 * 3 => 22

2 * 4 + 3 * 4 + 2 * 1 => 22

2 * 3 + 3 * 1 + 2 * 1 => 11

 

4 * 5 + 2 * 2 + 4 * 3 => 36

4 * 4 + 2 * 4 + 4 * 1 => 28

4* 3 + 2 * 1 + 4 * 1 => 18

 

3 * 5 + 1 * 2 + 4* 3 => 29

3 * 4 + 1 * 4 + 4 * 1 => 20

3 * 3 + 1 * 1 + 4 * 1 => 14

 

answer = [[22,22,11],[36,28,18],[29,29,18]]

 

이와 같이 진행되기 위해선

 

먼저 answer을 담을 배열을 만들어줍니다. (행 * 열 숫자들을 담는 것이므로 행의 갯수만큼 [Int]를 미리 만들어줍니다.->  var answer = Array(repeating: [Int](), count: arr1.count)

 

arr1을 맨 바깥 for문으로 반복해주고 -> for i in 0..<arr1.count {...}

 

arr2의 안의 숫자만큼 for문으로 반복해주고 -> for j in 0..<arr2[0].count{...}

 

answer에 추가될 숫자를 담을 변수를 만들어줍니다. ->var value = 0

 

arr2의 갯수만큼 반복해줍니다. -> for k in 0..<arr2.count{...}

 

value에 위에 설명한대로 값을 넣어줍니다. -> value += arr1[i][k] * arr2[k][j]

 

arr2의 갯수만큼 반복하는 for문을 나와 answer에 arr1의 차례대로 넣어줍니다. -> answer[i].append(value)

 

전체풀이

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

import Foundation   func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {     var answer = Array(repeating: [Int](), count: arr1.count)          for i in 0..<arr1.count {         for j in 0..<arr2[0].count{             var value = 0             for k in 0..<arr2.count{                 value += arr1[i][k] * arr2[k][j]             }             answer[i].append(value)         }     }     return answer } Colored by Color Scripter