C++ Algorithm & Study/C++ & Algorithm Strategies

[Programmers] 행렬의 곱셈

GameChoi 2023. 1. 23. 19:15

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

0. Headers

#include <vector>
using namespace std;

1. 알고리즘

 - 행렬의 곱셈은 배열1의 첫번 째 가로줄 * 배열2의 첫번째 세로줄을 곱하는 것임

   - 따라서 위의 식을 곱하고 배열에 넣으면 됨

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> v;
        for (int j = 0; j < arr2[0].size(); j++)
        {
            int temp = 0;
            for (int k = 0; k < arr1[0].size(); k++)
                temp += arr1[i][k] * arr2[k][j];
            v.push_back(temp);
        }
        answer.push_back(v);
    }
    return answer;
}

2. 완성 코드

#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> v;
        for (int j = 0; j < arr2[0].size(); j++)
        {
            int temp = 0;
            for (int k = 0; k < arr1[0].size(); k++)
                temp += arr1[i][k] * arr2[k][j];
            v.push_back(temp);
        }
        answer.push_back(v);
    }
    return answer;
}