프로그래머스 : 연습문제 > 행렬의 곱셈(lv2) c++

프로그래머스 링크

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

프로그래머스

문제 설명

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

제한 조건

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

입출력 예arr1arr2return

[[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;
    
    // 행렬의 곲의 규칙을 보면 좌행렬의 행의 길이와 우행렬 열의길이가 같기때문에
    // 3가지만 구해주면 된다.
    // arr1의 세로
    int Y1 = arr1.size();
    // arr2의 세로
    int Y2 = arr2.size();
    // arr2의 가로
    int X2 = arr2[0].size();
    
    answer.resize(arr1.size(), vector<int>(X2, 0));
    
    // 계산
    for(int i = 0; i < Y1; ++i)
    {
        for(int j = 0; j < X2; ++j)
        {
            for(int k = 0; k < Y2; ++k)
            {
                answer[i][j] += arr1[i][k] * arr2[k][j];
            }
        }
    }
    
    return answer;
}

정확성