본문 바로가기

알고리즘 · 코딩

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

문제 링크

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

 

코딩테스트 연습 - 행렬의 곱셈 | 프로그래머스

[[2, 3, 2], [4, 2, 4], [3, 1, 4]] [[5, 4, 3], [2, 4, 1], [3, 1, 1]] [[22, 22, 11], [36, 28, 18], [29, 20, 14]]

programmers.co.kr


C++ 풀이

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

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
	vector<vector<int>> answer;
	int arr2_Row_Size = arr2[0].size();
	for (int i = 0; i < arr1.size(); i++) {
		vector <int> x;
		for (int k = 0; k < arr2_Row_Size; k++) {
			int sum = 0;
			for (int j = 0; j < arr1[i].size(); j++) {
				sum += arr1[i][j] * arr2[j][k];
			}
			x.push_back(sum);
		}
		answer.push_back(x);
	}
	return answer;
}

행렬의 곱셈 과정을 그대로 적용시켜 풀었다.

 

 

 

반응형