코딩테스트/프로그래머스
[프로그래머스] 행렬의 곱셈
개발자마이
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;
}
}
반응형