본문 바로가기
[ 다먹살 ]/- Coding

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

by 엉망으로살기 2021. 7. 28.
반응형

https://programmers.co.kr/learn/courses/30/lessons/12949?language=java 

 

코딩테스트 연습 - 행렬의 곱셈

[[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]]

programmers.co.kr


사실 손으로 계산을 하거나 수식으로 썼을 때에는 이미 알고 있는 내용이라 금방 해결할 수 있을거라 생각했는데 생각보다 시간이 좀 걸렸다. 생각하는 걸 코드로 풀어내는 연습을 많이 해야할 것 같다. (배열, 인덱스)

 

 

 

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

반응형

댓글