[코딩테스트] 프로그래머스 12949, 행렬의 곱셈

2022. 7. 17. 16:37CodingTest

문제

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

 

프로그래머스

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

programmers.co.kr

 

문제접근

  • 행렬의 곱셈은 다음과 같습니다.

 

구현

class Solution {
	public int[][] solution(int[][] arr1, int[][] arr2) {
		int row = arr1.length;
		int col = arr1[0].length;
		int col2 = arr2[0].length;
		int[][] answer = new int[row][col2];

		for(int i = 0; i < row; i++){
			for(int j = 0; j < col2; j++){
				for(int k = 0; k < col; k++){
					answer[i][j] += arr1[i][k] * arr2[k][j];
				}
			}
		}
		return answer;
	}

	public static void main(String[] args){
		int[][] arr1 = {{1,4}, {3,2}, {4,1}};
		int[][] arr2 = {{3,3,3},{3,3,3}};

		new Solution().solution(arr1, arr2);
	}
}