마이의 개발 블로그

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

코딩테스트/프로그래머스

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

개발자마이 2023. 6. 23. 02:04
반응형
 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

접근 방식

 

행렬의 곱셈을 그대로 옮겨서 작성하면 되는 문제로 접근법이라고 할 게 없는 간단한 문제이다. 

로직 설명

- 답변 배열 선언(사이즈: arr1의 행 x arr2의 열)

- 답변 배열을 탐색하며

arr[i][j]에 곱셈 후 합산 결과값을 넣어줌(합산을 위해 위해 k를 사용)

Note

- 맞추긴 했는데 행렬 계산법이 헷갈려서 다시 한 번 개념을 찾아봤다. 왜 2단계 문제인지는 모르겠다.

정답코드

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