[프로그래머스] 행렬의 곱셈

최지나·2023년 11월 30일
1

코딩테스트

목록 보기
90/154

문제

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건
행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
곱할 수 있는 배열만 주어집니다.

입출력 예

arr1arr2return
[[1, 4], [3, 2], [4, 1]][[3, 3], [3, 3]][[15, 15], [15, 15], [15, 15]]
[[2, 3, 2], [4, 2, 4], [3, 1, 4]][[5, 4, 3], [2, 4, 1], [3, 1, 1]][[22, 22, 11], [36, 28, 18], [29, 20, 14]]

문제 출처

https://school.programmers.co.kr/learn/courses/30/lessons/12949

생각

  • 대학교 다닐 때 지겹게 손으로 계산하던 건데,, ✏️✏️ 코테 문제로 보니까 반갑고 손으로 풀었던게 억울했다(?)
  • answer[A][C] = arr1[A][B] * arr2[B][C]의 누적합. 행렬이 곱해질려면 B는 동일할 수 밖에 없음. B = arr1의 가로 길이 = arr2의 세로 길이

코드

class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
        
        int[][] answer = new int[arr1.length][arr2[0].length];
        
        for (int i = 0; i < arr1.length; i++){
            for (int j = 0; j < arr2[0].length; j++){
                for (int w = 0; w < arr2.length; w++){
                    answer[i][j] += arr1[i][w] * arr2[w][j];
                }
            }
        }
        
        return answer;
    }
}

다른 사람의 풀이

  • 행렬의 곱셈을 구하는 공식이 있다보니 대부분의 사람들의 풀이가 동일했다 😊😊
profile
의견 나누는 것을 좋아합니다 ლ(・ヮ・ლ)

0개의 댓글