BOJ - 행렬 곱셈 2740

 

해결방법

먼저 동적으로 이차원 배열을 입력받아야 하기 때문에 이차원 벡터를 선언해줘야 한다.

행렬을 입력받는것은 쉬우니 생략한다.

행렬 곱을 하는 과정은 다음과 같다. 첫번째 행렬의 첫 번째 행의 숫자를 두번 째 행렬의 첫번째 열의 모든 숫자들과 곱하고 그 값을 더해준다.

 

행렬은 다음처럼 계산이 진행된다.

01234567891011

2x3배열과 3x2배열의 행렬 곱 계산

따라서 우리가 해줘야 할 것은 행렬 곱의 크기인 N*K 사이즈의 배열을 만들고, 위의 그림처럼 각 배열마다 하나씩 연산을 더해주는 부분이다.

 

정답 코드

#include <iostream>
#include <vector>

using namespace std;

int main()
{
	int N, M;

	cin >> N >> M;

	vector<vector <int>> A(N, vector<int>(M));

	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < M; j++)
		{
			cin >> A[i][j];
		}
	}

	int C, K;

	cin >> C >> K;

	vector<vector <int>> B(C, vector<int>(K));

	for (int i = 0; i < C; i++)
	{
		for (int j = 0; j < K; j++)
		{
			cin >> B[i][j];
		}
	}

	vector<vector<int>> result(N, vector<int>(K));

	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < K; j++)
		{
			for (int k = 0; k < M; k++)
			{
				result[i][j] += A[i][k] * B[k][j];
			}
		}
	}

	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < K; j++)
		{
			cout << result[i][j] << " ";
		}
		cout << endl;
	}

	return 0;
}