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

 

🔗 문제 링크

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

✍🏻 문제 설명

💁🏻‍♀️ 내 풀이

행렬의 곱셈

행렬의 곱셈 문제이다. 행렬 곱셈의 방법은 알아야 풀이하기 쉽다. (X자 기억하시나요?)

 

나머지는 아래 풀이를 보면 이해가 갈 것이고, 이번에 프로그래머스에서 처음 2차원 answer를 어떻게 초기화 하는지 GPT가 알려줬다.

 

결과 행렬 초기화 부분을 보면 되는데, 각각 벡터 값이 어떻게 될지 알 수 있을 땐 이렇게 초기화해서 사용하면 좋을 것 같다.

 

🤖 코드

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    
    int n = arr1.size();    // 행렬 A의 행 개수
    int m = arr2[0].size(); // 행렬 B의 열 개수
    int k = arr1[0].size(); // 행렬 A의 열 개수 == 행렬 B의 행 개수

    vector<vector<int>> answer(n, vector<int>(m, 0)); // 결과 행렬 초기화

    for (int i = 0; i < n; i++) {        // 행렬 A의 행 반복
        for (int j = 0; j < m; j++) {    // 행렬 B의 열 반복
            for (int l = 0; l < k; l++) { // 행렬 곱셈 수행
                answer[i][j] += arr1[i][l] * arr2[l][j];
            }
        }
    }
    
    return answer;
}