프로그래머스 코딩테스트 연습 -행렬의 곱셈 LV2: (C++)

1.문제

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건

• 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
• 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
• 곱할 수 있는 배열만 주어집니다.

 

2.입력

arr1	arr2	return
[[1, 4], [3, 2], [4, 1]]	[[3, 3], [3, 3]]	[[15, 15], [15, 15], [15, 15]]
[[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]]

 

소스코드

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

 

 

체점 결과

정확성: 100.0

합계: 100.0 / 100.0

 

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