프로그래머스 문제풀이/LEVEL 2

[C++] 프로그래머스 문제풀이 LEVEL 2 행렬의 곱셈

코딩준우 2023. 7. 3. 01:40

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/12949?language=cpp 

 

프로그래머스

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

programmers.co.kr

 

 

#include <string>
#include <vector>

using namespace std;

int res(int i, int j, vector<vector<int>> & arr1, vector<vector<int>> & arr2){
    int result = 0;
    for (int k = 0; k < arr1[0].size(); ++k) {
        result += (arr1[i][k] * arr2[k][j]);
    }
    return result;
}

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    int row = arr1.size();
    int col = arr2[0].size();
    vector<vector<int>> answer(row, std::vector<int>(col, 0));

    for (int i = 0; i < row; ++i){
        for (int j = 0; j < col; ++j){
            answer[i][j] = res(i, j, arr1, arr2);
        }
    }
    return answer;
}