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

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

 

 

 

 

 

 

행렬의 덧셈은 행의 개수와 열의 개수가 같은 것끼리만 가능하다.

 

 

 

** Javascript

 

나의 풀이는 이렇다.

 

function solution(arr1, arr2) {
    return arr1.map((col, idx1) => col.map((row, idx2) => row + arr2[idx1][idx2]));
}

 

당시 javascript 의 다양한 기능들을 몰라서 map만 이용하여 이렇게 풀었다. 그런데 나 말고 다른 사람들도 생각이 비슷했던 것 같다.

 

근데 정확히는 col 과 row 위치가 바뀌어야 한다 ㅎㅎ

 

 

** Java

 

나의 풀이는 이렇다.

 

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

 

 

arr1[0].length 를 쓴 부분이 찝찝했다.

 

그런데 문제 제출 후 다른 사람들의 풀이를 보니 다 같은 생각이었던 것 같다. 추천을 두번째로 많이 받은 풀이는 내것과 거의 같았고 가장 많이 추천을 받은 풀이도 결국 arr1[0].length 를 사용하였다.