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

Seulguo·2022년 10월 6일
0

Algorithm

목록 보기
181/185
post-thumbnail
post-custom-banner

🐣 문제

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


🐤 풀이

• 2차원 벡터를 초기화 하는 방법

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

🐥 코드

#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;
}
post-custom-banner

0개의 댓글