백준 2740 ( C++ )

간단한 행렬 곱셈 문제

 

정수 N , M , K , 행렬의 원소들을 입력받은 후 행렬 곱셈을 수행하면 되는 문제.

 

( N , M ) 행렬과 ( M, K ) 행렬을 곱하면  ( N , K ) 행렬이 나오고,

 

행렬의 곱셈식을 이용하여 코드를 생성하면된다.

 

 

#include <iostream>
 
using namespace std;
 
const int MAX = 100;
 
int A[MAX][MAX];
int B[MAX][MAX];
int result[MAX][MAX];
int N = 0, M = 0, K = 0;
 
void multiple(int i, int j , int m);
 
int main(void) {
 
    int N, M, K;
 
    cin >> N >> M;
 
    for (int i = 0; i < N; i++)
        for (int j = 0; j < M; j++)
            cin >> A[i][j];
 
    cin >> M >> K;
 
    for (int i = 0; i < M; i++)
        for (int j = 0; j < K; j++)
            cin >> B[i][j];
 
    int m = M;
 
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < K; j++) {
            multiple(i, j , m);            
        }        
    }
    
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < K; j++) {
            cout << result[i][j] << " ";
        }
        cout << endl;
    }
    
}
 
 
void multiple(int i, int j, int m) {
    int sum = 0;
    for (int a = 0; a < m; a++) {
        sum += A[i][a] * B[a][j];
        
    }
    
    result[i][j] = sum;
}