행렬의 덧셈

문제

---

나의 생각

2차원 배열의 길이에 대한 이해도를 물어 보는 문제이다. arr[i].length가 1인경우에만 바로 계산을 하여 입력하도록 if문 입력 하는 방식으로 접근

---

나의 답안

public static int[][] solution(int[][] arr1, int[][] arr2) {
    int[][] answer = new int[arr1.length][arr1[0].length];

    for (int i = 0; i < arr1.length; i++) {
        if (arr1[i].length == 1) {
            answer[i][0] = arr1[i][0] + arr2[i][0];
        }
        for (int j = 0; j < arr1[i].length; j++) {
            answer[i][j] = arr1[i][j] + arr2[i][j];
        }
    }

    return answer;

}