Swift랑 친해지기/programmers 풀기

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

데브킹덕 2022. 10. 14. 21:12

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

 

 

나만의 풀이 

행렬끼리의 곱셈 성질을 이용해 for loop 반복문을 통해 arr1요소와  arr2요소를 곱하도록 하였다.

sumArray로 곱한 행렬의 1행씩을 저장하였고 행이 바뀔때 removeAll()메서드를 이용해 제거하였고

곱한 행렬을 resultArray 2차원 배열에 저장하도록함

 

2차원 Int자료형 빈배열 

var 이름 = [[Int]]() 

import Foundation

func solution(_ arr1:[[Int]], _ arr2:[[Int]]) -> [[Int]] {
    
    var sumArray = [Int]()
    var resultArray = [[Int]]()
    for i in 0..<arr1.count{
        for k in 0..<arr2[0].count{
            var sum = 0
            for p in 0..<arr2.count{
                sum += arr1[i][p] * arr2[p][k]
            }
            sumArray.append(sum)
        }
        resultArray.append(sumArray)
        sumArray.removeAll()
    }
    return resultArray
}