C++ / 프로그래머스 / 행렬의 곱셈

문제

 

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

프로그래머스

문제풀이

 

곱셈 방법은 알지만 식으로 어떻게 나타낼지 몰라서 고민을 좀 했다.

먼저 x*y과 n*m 벡터가 있으면 y와 n은 같아야 하고 최종적인 크기는 x*m 크기의 행렬이 된다.

 

새로 만들 벡터는 y크기는 arr1의 크기, x크기는 arr2의 가로 크기인 arr2[숫자]의 크기로 두면 된다.

크기가 넉넉한 행렬 1과 행렬 2가 있으면 행렬 1에서는 가로방향으로, 행렬 2에서는 세로방향 순서로 곱셈을 하게 된다.

그다음은 행렬 1은 그대로, 행렬 2에서는 x+1인 위치에서 세로방향으로 곱셈을 하게 된다.

 

행렬 1의 x는 0, 1만 반복하고, 행렬 2의 y는 0, 1만 반복하고 x는 1씩 증가한다.

이것을 삼중 반복문을 통해 코드를 작성하면 된다.

코드

#include <string>
#include <vector>
#include <iostream>
using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    int y(0), x(0), idx(0), idx2(0);
    vector<vector<int>>arr3(arr1.size(), vector<int>(arr2[0].size(), 0));

    for (y = 0; y < arr3.size(); ++y) {
        for (x = 0; x < arr3[0].size(); ++x) {
            for (idx = 0; idx < arr2.size(); ++idx) {
                arr3[y][x] += (arr1[y][idx] * arr2[idx][x]);
            }
        }
    }

    return arr3;
}