백준 2740 행렬 곱셈(Java)

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

2740번: 행렬 곱셈

---

 

해석 및 팁

 

---

 

이 문제를 풀려면 행렬곱셈의 정의를 알고 있어야 합니다. 간단히 설명하면 아래의 사진과 같습니다.

https://ko.wikipedia.org/wiki/%ED%96%89%EB%A0%AC_%EA%B3%B1%EC%85%88

행렬 곱셈 - 위키백과, 우리 모두의 백과사전

먼저 반복문의 통해 A행렬과 B행렬의 값을 저장한후 공식에 따라 StringBuilder에 저장한 후에 출력해 주면 됩니다.

---

 

Java 코드

 

---

import java.util.Scanner;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    StringBuilder sb = new StringBuilder();
   
    int N = sc.nextInt();
    int M = sc.nextInt();
    int[][] a = new int[N][M];

    for(int i = 0; i < N; i++) {
      for(int j = 0; j < M; j++) {
        a[i][j] = sc.nextInt();
      }
    }

    M = sc.nextInt();
    int K = sc.nextInt();
    int[][] b = new int[M][K];
    
     for(int i = 0; i < M; i++) {
      for(int j = 0; j < K; j++) {
        b[i][j] = sc.nextInt();
      }
    }
    
    for(int i = 0; i < N; i++) {
      for(int j = 0; j < K; j++) {
        int sum = 0;
        for(int k = 0; k < M; k++) {
          sum += a[i][k] * b[k][j];
        }
        sb.append(sum).append(" ");
      }
      sb.append('\n');
    }

    System.out.println(sb);
  }
}