[LV1/JAVA] 행렬의 덧셈

 

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

프로그래머스

 

 

1. 예전에 학교에서 C언어 배울 때 이후로 오랜만에 보는 2차원 배열... 실무에서는 거의 쓸 일이 없었다.

2. 헷갈려서 배열 요소에 대한 그림을 그리며 풀었다.

 

나의 풀이

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;
    }
}

 

 

처음에 answer 배열을 선언할 때 배열의 크기를 아래와 같이 했더니 테스트 케이스 2를 통과하지 못했다.

int[][] answer = new int[arr1.length][arr2.length];

 

 

테스트 케이스 2에서 [[4], [6]] 이렇게 출력이 돼야 하는데 [[4,0],[6,0]] 이런식으로 실행결과가 나왔다.

아래와 같이 열의 크기를 행의 크기와 맞춰줘야 기댓값대로 출력할 수 있다.

int[][] answer = new int[arr1.length][arr1[0].length];