PS/Programmers

Programmers / Level 2 / 행렬의 곱셈 / C++

KimMinJun 2022. 2. 11. 19:48
문제 설명

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

 

제한 조건
  • 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
  • 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
  • 곱할 수 있는 배열만 주어집니다.
입출력 예
arr1 arr2 return
[[1, 4], [3, 2], [4, 1]] [[3, 3], [3, 3]] [[15, 15], [15, 15], [15, 15]]
[[2, 3, 2], [4, 2, 4], [3, 1, 4]] [[5, 4, 3], [2, 4, 1], [3, 1, 1]] [[22, 22, 11], [36, 28, 18], [29, 20, 14]]

 

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;

    for(int i=0; i<arr1.size(); i++) {
        vector<int> tmp;
        for(int j=0; j<arr2[0].size(); j++) {
            int sum = 0;
            for(int k=0; k<arr2.size(); k++) {
                sum += arr1[i][k] * arr2[k][j];
            }
            tmp.push_back(sum);
        }
        answer.push_back(tmp);
    }

    return answer;
}

행렬의 기본 곰셈식만 알고 있으면 반복문으로 간단히 나타낼 수 있다.

처음에 코드로 옮길때 인덱스를 어떻게 설정해야 할지 헷갈리지만 직접 그리면서 하다보면 쉽게 이해할 수 있다.