[BOJ] 백준 2740 행렬 곱셈

Approach

출처: https://www.acmicpc.net/problem/2740

행렵 곱셈을 구현하는 문제이다.

 

행렬 곱셈을 위해서는 (N × M) * (

#include <iostream>

 

using namespace std;

 

const int MAX_SIZE = 100 + 2;

int A[MAX_SIZE][MAX_SIZE], B[MAX_SIZE][MAX_SIZE], result[MAX_SIZE][MAX_SIZE];

int N, M, K;

 

int main()

{

    scanf("%d %d", &N, &M);

    for (int i = 0; i < N; i++)

    {

        for (int j = 0; j < M; j++)

        {

            scanf("%d", &A[i][j]);

        }

    }

 

    scanf("%d %d", &M, &K);

    for (int i = 0; i < M; i++)

    {

        for (int j = 0; j < K; j++)

        {

            scanf("%d", &B[i][j]);

        }

    }

 

    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]);

            }

            printf("%d ", result[i][j]);

        }

        puts("");

    }

}

 × K)  → (N × K)

연산되는 순서에 맞춰 for문 구현하면 된다.

---

#include <iostream>

using namespace std;

const int MAX_SIZE = 100 + 2;
int A[MAX_SIZE][MAX_SIZE], B[MAX_SIZE][MAX_SIZE], result[MAX_SIZE][MAX_SIZE];
int N, M, K;

int main()
{
    scanf("%d %d", &N, &M);
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < M; j++)
        {
            scanf("%d", &A[i][j]);
        }
    }

    scanf("%d %d", &M, &K);
    for (int i = 0; i < M; i++)
    {
        for (int j = 0; j < K; j++)
        {
            scanf("%d", &B[i][j]);
        }
    }

    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]);
            }
            printf("%d ", result[i][j]);
        }
        puts("");
    }
}