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

행렬의 곱셈 / Lv.2

---

문제  설명 )

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

 

 

 

제한 사항 )

• 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
• 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
• 곱할 수 있는 배열만 주어집니다.

 

 

 

입출력 예 )

 

 

 

 

풀이)

 

#include <string>
#include <vector>
 
using namespace std;
 
vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;
    
    answer.resize(arr1.size());
    
    // 행렬1의 가로
    for(int i = 0; i < arr1.size(); i++)
    {
        // 행렬 2의 세로
        for(int j = 0; j < arr2[0].size(); j++)
        {
            int temp = 0;
            // 곱하는 인덱스
            for(int k = 0; k < arr1[0].size(); k++)
            {
                temp += (arr1[i][k] * arr2[k][j]);
            }
            answer[i].push_back(temp);
        }
    }
    
    return answer;
}

---

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

프로그래머스