포스트

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

문제


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

풀이


단순한 행렬곱 문제로, arr1의 행과, arr2의 열 크기의 2차원 벡터 안에 arr1의 행과 arr2의 열 간의 곱을 각각 곱한 후, 더해줘 값을 입력한다.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) 
{
    vector<vector<int>> answer(arr1.size(), vector <int>(arr2[0].size(), 0));

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

    return answer;
}

이 기사는 저작권자의 CC BY 4.0 라이센스를 따릅니다.