[자바] 백준 2740 - 행렬 곱셈 (java)

 문제 : boj2740

---

 

필요 알고리즘 개념

• 구현, 수학
  
  • 행렬 곱셈하는 방법을 알고 있다는 전제하에, 단순 구현문제이다.

※ 제 코드에서 왜 main 함수에 로직을 직접 작성하지 않았는지, 왜 Scanner를 쓰지 않고 BufferedReader를 사용했는지 등에 대해서는 '자바로 백준 풀 때의 팁 및 주의점' 글을 참고해주세요. 백준을 자바로 풀어보려고 시작하시는 분이나, 백준에서 자바로 풀 때의 팁을 원하시는 분들도 보시는걸 추천드립니다.

 

---

 

풀이

  혹시 행렬 곱셈 방법을 모른다면 애초에 이 문제를 풀 수 없으니 구글링을 통해 행렬 곱셈방법을 알아보자. 알고있다면, 이제 구현력(?)에 달려있다. 두 행렬을 입력받은 후 행렬 곱셈을 해주는 코드를 작성하면 되며, 구현방식을 잘 모르겠다면 이하 코드의 matrixMult()를 참고하자.

 

 

---

 

코드 : github

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    private int[][] matrixMult(int[][] a, int[][] b) {
        int r = a.length;
        int c = b[0].length;
        int[][] arr = new int[r][c];
        for (int i = 0; i < r; i++) {
            for (int j = 0; j < c; j++) {

                for (int k = 0; k < b.length; k++) {
                    arr[i][j] += a[i][k] * b[k][j];
                }

            }
        }
        return arr;
    }

    private void solution() throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());
        int[][] a = new int[n][m];
        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < m; j++) {
                a[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        st = new StringTokenizer(br.readLine());
        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());
        int[][] b = new int[n][m];
        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < m; j++) {
                b[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        int[][] answer = matrixMult(a, b);
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < answer.length; i++) {
            for (int j = 0; j < answer[0].length; j++) {
                sb.append(answer[i][j]).append(' ');
            }
            sb.append('\n');
        }
        System.out.print(sb);
    }

    public static void main(String[] args) throws Exception {
        new Main().solution();
    }
}